Mirrors
/
EveryDream2trainer
mirror of https://github.com/victorchall/EveryDream2trainer.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

If there are no tags just return the main prompt

This commit is contained in:
Joel Holdbrooks 2023-01-23 11:27:12 -08:00
parent c1a66317cd
commit 646f383188
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
data/image_train_item.py
View File

@ -74,7 +74,7 @@ class ImageCaption:
tags_caption = self.__get_shuffled_tags(seed, self.__tags) 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 + ", " + tags_caption return self.__main_prompt
def get_caption(self) -> str: def get_caption(self) -> str:
if self.__tags: if self.__tags: