Update train_util.py

kohya-ss · Oct 5, 2024 · 659cf31 · 659cf31
1 parent 77d4e1f
commit 659cf31
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/library/train_util.py b/library/train_util.py
@@ -19,6 +19,7 @@
     Sequence,
     Tuple,
     Union,
+    Callable,
 )
 from accelerate import Accelerator, InitProcessGroupKwargs, DistributedDataParallelKwargs, PartialState
 import glob