From 05144052be95b7bd64425d9ef75d43cd64ee03fc Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Tue, 17 Jan 2023 11:50:39 +0100
Subject: Fix

---
 training/strategy/ti.py | 1 -
 1 file changed, 1 deletion(-)

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(
         on_train=on_train,
         on_eval=on_eval,
         on_after_optimize=on_after_optimize,
-        on_after_epoch=on_after_epoch,
         on_log=on_log,
         on_checkpoint=on_checkpoint,
         on_sample=on_sample,
-- 
cgit v1.2.3-70-g09d2