diff --git a/README.md b/README.md index 2d2e9982..20e9ba52 100644 --- 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 index 6af402f8..63f54ea9 100644 --- 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 index 2eaaf8f8..d7dfdb15 100644 --- 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" \ No newline at end of file diff --git a/docs/reference/config/shared/optimizer_config.md b/docs/reference/config/shared/optimizer_config.md index ada81797..f2abc909 100644 --- 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 index 3fe24fc5..a4c4338a 100644 --- 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: