diff options
Diffstat (limited to 'training/strategy')
-rw-r--r-- | training/strategy/ti.py | 1 |
1 files changed, 0 insertions, 1 deletions
diff --git a/training/strategy/ti.py b/training/strategy/ti.py index eb6730b..1af834b 100644 --- a/training/strategy/ti.py +++ b/training/strategy/ti.py | |||
@@ -154,7 +154,6 @@ def textual_inversion_strategy_callbacks( | |||
154 | on_train=on_train, | 154 | on_train=on_train, |
155 | on_eval=on_eval, | 155 | on_eval=on_eval, |
156 | on_after_optimize=on_after_optimize, | 156 | on_after_optimize=on_after_optimize, |
157 | on_after_epoch=on_after_epoch, | ||
158 | on_log=on_log, | 157 | on_log=on_log, |
159 | on_checkpoint=on_checkpoint, | 158 | on_checkpoint=on_checkpoint, |
160 | on_sample=on_sample, | 159 | on_sample=on_sample, |