From 307099c4875037d8e38fbce6c9bcfae53affc0bf Mon Sep 17 00:00:00 2001
From: DKnight54 <126916963+DKnight54@users.noreply.github.com>
Date: Fri, 31 Jan 2025 03:21:37 +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 4c7b0b41..83ae5122 100644
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -5520,7 +5520,7 @@ def sample_images_common(
         
         if '__caption__' in prompt_dict.get("prompt") and example_tuple:
             logger.info(f"Original prompt: {prompt_dict.get('prompt')}")
-            if len(example_tuple[1] > 1:
+            if len(example_tuple[1]) > 1:
                 while example_tuple[1][idx] == '':
                     idx = (idx + 1) % len(example_tuple[1])
                     if idx == 0: