diff options
Diffstat (limited to 'train_dreambooth.py')
-rw-r--r-- | train_dreambooth.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/train_dreambooth.py b/train_dreambooth.py index 4d1e0a3..c355ea1 100644 --- a/train_dreambooth.py +++ b/train_dreambooth.py | |||
@@ -638,7 +638,7 @@ def main(): | |||
638 | if args.train_text_encoder: | 638 | if args.train_text_encoder: |
639 | print(f"Training entire text encoder.") | 639 | print(f"Training entire text encoder.") |
640 | 640 | ||
641 | embeddings.make_permanent() | 641 | embeddings.persist() |
642 | text_encoder.text_model.embeddings.temp_token_embedding.requires_grad_(False) | 642 | text_encoder.text_model.embeddings.temp_token_embedding.requires_grad_(False) |
643 | else: | 643 | else: |
644 | print(f"Training added text embeddings") | 644 | print(f"Training added text embeddings") |