minor fix

2026-03-18 22:08:13 +00:00 · 2025-08-06 10:58:53 +08:00
parent 4299c999b5
commit 3915bc3ee6
2 changed files with 2 additions and 1 deletions
--- a/diffsynth/trainers/utils.py
+++ b/diffsynth/trainers/utils.py
@@ -404,6 +404,7 @@ def launch_training_task(
    )
    model, optimizer, dataloader, scheduler = accelerator.prepare(model, optimizer, dataloader, scheduler)
    
+    global_steps = 0
    for epoch_id in range(num_epochs):
        for step_id, data in enumerate(tqdm(dataloader)):
            with accelerator.accumulate(model):