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

Fix caption loading.

Browse Source
This commit is contained in:
Hameer Abbasi
2025-06-11 14:47:26 +02:00
parent ffb5fe0667
commit c5d6b74fea
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
toolkit/dataloader_mixins.py
View File

@@ -315,7 +315,7 @@ class CaptionProcessingDTOMixin:
# see if prompt file exists # see if prompt file exists
path_no_ext = os.path.splitext(self.path)[0] path_no_ext = os.path.splitext(self.path)[0]
prompt_ext = self.dataset_config.caption_ext prompt_ext = self.dataset_config.caption_ext
prompt_path = f"{path_no_ext}.{prompt_ext}" prompt_path = path_no_ext + prompt_ext
short_caption = None short_caption = None
if os.path.exists(prompt_path): if os.path.exists(prompt_path):