diff options
| -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, |