Merge pull request #120 from invoke-ai/docs-fixes

Minor fixes to docs and logging
invoke-ai · May 3, 2024 · b1baf9a · b1baf9a
2 parents f00f912 + 56750a2
commit b1baf9a
Show file tree

Hide file tree

Showing 5 changed files with 13 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -2,7 +2,8 @@
 
 A library for training custom Stable Diffusion models (fine-tuning, LoRA training, textual inversion, etc.) that can be used in [InvokeAI](https://github.com/invoke-ai/InvokeAI).
 
-> [!WARNING] > `invoke-training` is still under active development, and breaking changes are likely. Full backwards compatibility will not be guaranteed until v1.0.0.
+> [!WARNING]
+> `invoke-training` is still under active development, and breaking changes are likely. Full backwards compatibility will not be guaranteed until v1.0.0.
 > In the meantime, I recommend pinning to a specific commit hash.
 
 ## Documentation

diff --git a/docs/reference/config/shared/data/data_loader_config.md b/docs/reference/config/shared/data/data_loader_config.md
@@ -1 +1,4 @@
 ::: invoke_training.config.data.data_loader_config
+    options:
+      filters:
+      - "!^model_config"
diff --git a/docs/reference/config/shared/data/dataset_config.md b/docs/reference/config/shared/data/dataset_config.md
@@ -1 +1,4 @@
 ::: invoke_training.config.data.dataset_config
+    options:
+      filters:
+      - "!^model_config"
diff --git a/docs/reference/config/shared/optimizer_config.md b/docs/reference/config/shared/optimizer_config.md
@@ -1 +1,4 @@
 ::: invoke_training.config.optimizer.optimizer_config
+    options:
+      filters:
+      - "!^model_config"
diff --git a/src/invoke_training/_shared/stable_diffusion/validation.py b/src/invoke_training/_shared/stable_diffusion/validation.py
@@ -76,7 +76,7 @@ def generate_validation_images_sd(  # noqa: C901
             negative_prompt = None
             if config.negative_validation_prompts is not None:
                 negative_prompt = config.negative_validation_prompts[prompt_idx]
-            logger.info(f"Validation prompt {prompt_idx}, pos: '{positive_prompt}', neg: '{negative_prompt}'")
+            logger.info(f"Validation prompt {prompt_idx}, pos: '{positive_prompt}', neg: '{negative_prompt or ''}'")
 
             generator = torch.Generator(device=accelerator.device)
             if config.seed is not None:
@@ -193,7 +193,7 @@ def generate_validation_images_sdxl(  # noqa: C901
             negative_prompt = None
             if config.negative_validation_prompts is not None:
                 negative_prompt = config.negative_validation_prompts[prompt_idx]
-            logger.info(f"Validation prompt {prompt_idx}, pos: '{positive_prompt}', neg: '{negative_prompt}'")
+            logger.info(f"Validation prompt {prompt_idx}, pos: '{positive_prompt}', neg: '{negative_prompt or ''}'")
 
             generator = torch.Generator(device=accelerator.device)
             if config.seed is not None: