summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorVolpeon <git@volpeon.ink>2023-01-17 11:50:39 +0100
committerVolpeon <git@volpeon.ink>2023-01-17 11:50:39 +0100
commit05144052be95b7bd64425d9ef75d43cd64ee03fc (patch)
treebc36169118f59252016935ee1e9aa0167ca77401
parentMake embedding decay work like Adam decay (diff)
downloadtextual-inversion-diff-05144052be95b7bd64425d9ef75d43cd64ee03fc.tar.gz
textual-inversion-diff-05144052be95b7bd64425d9ef75d43cd64ee03fc.tar.bz2
textual-inversion-diff-05144052be95b7bd64425d9ef75d43cd64ee03fc.zip
Fix
-rw-r--r--training/strategy/ti.py1
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,