diff options
| author | Volpeon <git@volpeon.ink> | 2023-03-23 11:07:57 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-03-23 11:07:57 +0100 |
| commit | 0767c7bc82645186159965c2a6be4278e33c6721 (patch) | |
| tree | a136470ab85dbb99ab51d9be4a7831fe21612ab3 /train_lora.py | |
| parent | Fix (diff) | |
| download | textual-inversion-diff-0767c7bc82645186159965c2a6be4278e33c6721.tar.gz textual-inversion-diff-0767c7bc82645186159965c2a6be4278e33c6721.tar.bz2 textual-inversion-diff-0767c7bc82645186159965c2a6be4278e33c6721.zip | |
Update
Diffstat (limited to 'train_lora.py')
| -rw-r--r-- | train_lora.py | 3 |
1 files changed, 0 insertions, 3 deletions
diff --git a/train_lora.py b/train_lora.py index 787f271..8dd3c86 100644 --- a/train_lora.py +++ b/train_lora.py | |||
| @@ -424,9 +424,6 @@ def main(): | |||
| 424 | tokenizer, text_encoder, vae, unet, noise_scheduler, sample_scheduler, embeddings = get_models( | 424 | tokenizer, text_encoder, vae, unet, noise_scheduler, sample_scheduler, embeddings = get_models( |
| 425 | args.pretrained_model_name_or_path) | 425 | args.pretrained_model_name_or_path) |
| 426 | 426 | ||
| 427 | tokenizer.set_use_vector_shuffle(args.vector_shuffle) | ||
| 428 | tokenizer.set_dropout(args.vector_dropout) | ||
| 429 | |||
| 430 | vae.enable_slicing() | 427 | vae.enable_slicing() |
| 431 | vae.set_use_memory_efficient_attention_xformers(True) | 428 | vae.set_use_memory_efficient_attention_xformers(True) |
| 432 | unet.enable_xformers_memory_efficient_attention() | 429 | unet.enable_xformers_memory_efficient_attention() |