Update train_network.py

2026-04-14 16:22:28 +00:00 · 2025-02-01 02:06:24 +08:00
parent c14aa5bf33
commit 13f764dbbe
1 changed files with 1 additions and 0 deletions
--- a/train_network.py
+++ b/train_network.py
@@ -39,6 +39,7 @@ from library.custom_train_functions import (
    apply_masked_loss,
 )
 from library.utils import setup_logging, add_logging_arguments
+from accelerate.utils import gather_object

 setup_logging()
 import logging