From 99bc4b6f48c840bda9d5c9a25d4dff30aca85833 Mon Sep 17 00:00:00 2001
From: DKnight54 <126916963+DKnight54@users.noreply.github.com>
Date: Thu, 30 Jan 2025 02:38:19 +0800
Subject: [PATCH] Update train_util.py

---
 library/train_util.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/library/train_util.py b/library/train_util.py
index 665b7143..d4a0ec45 100644
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -5506,6 +5506,7 @@ def sample_images_common(
     os.makedirs(save_dir, exist_ok=True)
 
     # preprocess prompts
+    idx = 0
     for i in range(len(prompts)):
         prompt_dict = prompts[i]
         if isinstance(prompt_dict, str):
@@ -5538,7 +5539,6 @@ def sample_images_common(
     if distributed_state.num_processes <= 1:
         # If only one device is available, just use the original prompt list. We don't need to care about the distribution of prompts.
         with torch.no_grad():
-            idx = 0
             for prompt_dict in prompts:
                 sample_image_inference(
                     accelerator, args, pipeline, save_dir, prompt_dict, epoch, steps, prompt_replacement, controlnet=controlnet