diff options
| author | brkirch <brkirch@users.noreply.github.com> | 2022-10-04 08:26:54 +0000 |
|---|---|---|
| committer | brkirch <brkirch@users.noreply.github.com> | 2022-10-04 08:26:54 +0000 |
| commit | 2adb2497401d164ce23edecdf01500c4f1fcfe91 (patch) | |
| tree | 1f74e02bc3bbc87a698a254941cf4839829dbe79 /modules/textual_inversion/preprocess.py | |
| parent | eeab7aedf532680a6ae9058ee272450bb07e41eb (diff) | |
| parent | 35a00b013007c16c97ed79bee4cefcb798181dea (diff) | |
| download | stable-diffusion-webui-gfx803-2adb2497401d164ce23edecdf01500c4f1fcfe91.tar.gz stable-diffusion-webui-gfx803-2adb2497401d164ce23edecdf01500c4f1fcfe91.tar.bz2 stable-diffusion-webui-gfx803-2adb2497401d164ce23edecdf01500c4f1fcfe91.zip | |
Merge branch 'cpu-cmdline-opt' of https://github.com/brkirch/stable-diffusion-webui into cpu-cmdline-opt
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
|