diff options
| author | Volpeon <git@volpeon.ink> | 2023-04-21 11:43:50 +0200 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-04-21 11:43:50 +0200 |
| commit | 7da4f0485032bb8b8acfc678546ffcea3a23a44b (patch) | |
| tree | 1e7880189df21132861114b5dbf4c614405c9855 /training/strategy/ti.py | |
| parent | Fix PTI (diff) | |
| download | textual-inversion-diff-7da4f0485032bb8b8acfc678546ffcea3a23a44b.tar.gz textual-inversion-diff-7da4f0485032bb8b8acfc678546ffcea3a23a44b.tar.bz2 textual-inversion-diff-7da4f0485032bb8b8acfc678546ffcea3a23a44b.zip | |
Update
Diffstat (limited to 'training/strategy/ti.py')
| -rw-r--r-- | training/strategy/ti.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/training/strategy/ti.py b/training/strategy/ti.py index f330cb7..6bc1d7d 100644 --- a/training/strategy/ti.py +++ b/training/strategy/ti.py | |||
| @@ -90,7 +90,7 @@ def textual_inversion_strategy_callbacks( | |||
| 90 | return nullcontext() | 90 | return nullcontext() |
| 91 | 91 | ||
| 92 | @contextmanager | 92 | @contextmanager |
| 93 | def on_train(epoch: int): | 93 | def on_train(cycle: int): |
| 94 | text_encoder.train() | 94 | text_encoder.train() |
| 95 | tokenizer.train() | 95 | tokenizer.train() |
| 96 | yield | 96 | yield |