From 646f3831888806f7fb0996d6ea866c8b90098486 Mon Sep 17 00:00:00 2001
From: Joel Holdbrooks <cjholdbrooks@gmail.com>
Date: Mon, 23 Jan 2023 11:27:12 -0800
Subject: [PATCH] If there are no tags just return the main prompt

---
 data/image_train_item.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/data/image_train_item.py b/data/image_train_item.py
index 6ce10fb..84f92de 100644
--- a/data/image_train_item.py
+++ b/data/image_train_item.py
@@ -74,7 +74,7 @@ class ImageCaption:
                 tags_caption = self.__get_shuffled_tags(seed, self.__tags)
 
             return self.__main_prompt + ", " + tags_caption
-        return self.__main_prompt + ", " + tags_caption
+        return self.__main_prompt
 
     def get_caption(self) -> str:
         if self.__tags: