add support for gelbooru tags in filenames for textual inversion

2026-04-26 17:29:09 +00:00 · 2022-10-04 08:52:11 +03:00
parent 2865ef4b9a
commit 5ef0baf5ea
2 changed files with 8 additions and 3 deletions
--- a/modules/textual_inversion/preprocess.py
+++ b/modules/textual_inversion/preprocess.py
@@ -26,7 +26,9 @@ def preprocess(process_src, process_dst, process_flip, process_split, process_ca
        if process_caption:
            caption = "-" + shared.interrogator.generate_caption(image)
        else:
-            caption = ""
+            caption = filename
+            caption = os.path.splitext(caption)[0]
+            caption = os.path.basename(caption)

        image.save(os.path.join(dst, f"{index:05}-{subindex[0]}{caption}.png"))
        subindex[0] += 1