diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-10-04 08:52:11 +0300 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-10-04 08:52:27 +0300 |
| commit | 5ef0baf5eaec7f21a1666af424405cbee19f3764 (patch) | |
| tree | 44ae1840049bc638ef602ff111d2fb25da1e533f /modules/textual_inversion/preprocess.py | |
| parent | 2865ef4b9ab16d56326cc805541bebcf01d099bc (diff) | |
add support for gelbooru tags in filenames for textual inversion
Diffstat (limited to 'modules/textual_inversion/preprocess.py')
| -rw-r--r-- | modules/textual_inversion/preprocess.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/modules/textual_inversion/preprocess.py b/modules/textual_inversion/preprocess.py index 209e928f..f545a993 100644 --- 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
|