diff options
| author | Volpeon <git@volpeon.ink> | 2023-01-05 22:05:25 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-01-05 22:05:25 +0100 |
| commit | 5c115a212e40ff177c734351601f9babe29419ce (patch) | |
| tree | a66c8c67d2811e126b52ac4d4cd30a1c3ea2c2b9 /train_dreambooth.py | |
| parent | Fix LR finder (diff) | |
| download | textual-inversion-diff-5c115a212e40ff177c734351601f9babe29419ce.tar.gz textual-inversion-diff-5c115a212e40ff177c734351601f9babe29419ce.tar.bz2 textual-inversion-diff-5c115a212e40ff177c734351601f9babe29419ce.zip | |
Added EMA to TI
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") |