diff options
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() |