From 5308a0bfc10268cbc8739627f1ac4d888a5486f8 Mon Sep 17 00:00:00 2001
From: DKnight54 <126916963+DKnight54@users.noreply.github.com>
Date: Sat, 1 Feb 2025 02:22:25 +0800
Subject: [PATCH] Update train_util.py

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

diff --git a/library/train_util.py b/library/train_util.py
index 9f8a6dea..44bba3f4 100644
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -5566,8 +5566,7 @@ def sample_images_common(
         for prompt in per_process_prompts:
             prompts.extend(prompt)
     distributed_state.wait_for_everyone()
-    per_process_prompts = gather_object(prompts)
-    prompts = []
+    prompts = gather_object(prompts)
                
     with torch.no_grad():
         with distributed_state.split_between_processes(prompts) as prompt_dict_lists: