Skip to content

Commit

Permalink
Merge pull request #125 from invoke-ai/docs-cleanup
Browse files Browse the repository at this point in the history
Fix merge conflict in SDXL finetuning + Docs cleanup
  • Loading branch information
RyanJDick committed May 14, 2024
2 parents 953b54a + e5718f9 commit 1f08662
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 2 deletions.
1 change: 1 addition & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,7 @@ https://invoke-ai.github.io/invoke-training/
- DreamBooth LoRA
- Textual Inversion
- Stable Diffusion XL
- Full finetuning
- LoRA
- DreamBooth LoRA
- Textual Inversion
Expand Down
4 changes: 2 additions & 2 deletions docs/tutorials/stable_diffusion/robocats_finetune_sdxl.md
Original file line number Diff line number Diff line change
Expand Up @@ -124,15 +124,15 @@ python src/invoke_training/scripts/_experimental/lora_extraction/extract_lora_fr
--lora-rank 32
```

## 5 - Import into InvokeAI
## 6 - Import into InvokeAI

If you haven't already, setup [InvokeAI](https://github.com/invoke-ai/InvokeAI) by following its documentation.

Import your finetuned diffusers model or your extracted LoRA from the 'Models' tab.

Congratulations, you can now use your new robocat model! 🎉

## 6 - Comparison: Finetune vs. LoRA Extraction
## 7 - Comparison: Finetune vs. LoRA Extraction

As noted earlier, the LoRA extraction process is lossy for a number of reasons. (There are some significant improvements coming to the LoRA extraction process soon to improve the fidelity of the extracted LoRA model.)

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -131,6 +131,7 @@ def train(config: SdxlFinetuneConfig, callbacks: list[PipelineCallbacks] | None

logger.info("Loading models.")
tokenizer_1, tokenizer_2, noise_scheduler, text_encoder_1, text_encoder_2, vae, unet = load_models_sdxl(
logger=logger,
model_name_or_path=config.model,
hf_variant=config.hf_variant,
vae_model=config.vae_model,
Expand Down

0 comments on commit 1f08662

Please sign in to comment.