github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Fixed bug in returning clip image embed to actually return it

Browse Source
This commit is contained in:
Jaret Burkett
2024-10-03 10:49:09 -06:00
parent 67e0aca750
commit 28e6f00790
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
toolkit/dataloader_mixins.py
View File

@@ -752,8 +752,7 @@ class ClipImageFileItemDTOMixin:
if len(img_files) > 1: if len(img_files) > 1:
img_files.remove(self.path) img_files.remove(self.path)
# randomly grab one # randomly grab one
self.clip_image_path = random.choice(img_files) return random.choice(img_files)
return self.path
else: else:
return self.clip_image_path return self.clip_image_path