From c5d6b74feafe5b2aa957c8407c4f806293432c5e Mon Sep 17 00:00:00 2001
From: Hameer Abbasi <2190658+hameerabbasi@users.noreply.github.com>
Date: Wed, 11 Jun 2025 14:47:26 +0200
Subject: [PATCH] Fix caption loading.

---
 toolkit/dataloader_mixins.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/toolkit/dataloader_mixins.py b/toolkit/dataloader_mixins.py
index 5f419ff0..4b1e86d4 100644
--- a/toolkit/dataloader_mixins.py
+++ b/toolkit/dataloader_mixins.py
@@ -315,7 +315,7 @@ class CaptionProcessingDTOMixin:
             # see if prompt file exists
             path_no_ext = os.path.splitext(self.path)[0]
             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
 
             if os.path.exists(prompt_path):