aboutsummaryrefslogtreecommitdiffstats
path: root/modules/textual_inversion/preprocess.py
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2023-06-27 06:23:15 +0000
committerGitHub <noreply@github.com>2023-06-27 06:23:15 +0000
commit0b97ae2832a89b103be1c062d7778e558732712f (patch)
tree6d0966156f1cfa78aa0e97ed0474afaffc50f2cd /modules/textual_inversion/preprocess.py
parent7e2d39a2d158d1426321686b05d31a3ea694a99e (diff)
parent3cd4fd51ef916aba8b978490569a5da82795a839 (diff)
downloadstable-diffusion-webui-gfx803-0b97ae2832a89b103be1c062d7778e558732712f.tar.gz
stable-diffusion-webui-gfx803-0b97ae2832a89b103be1c062d7778e558732712f.tar.bz2
stable-diffusion-webui-gfx803-0b97ae2832a89b103be1c062d7778e558732712f.zip
Merge branch 'dev' into master
Diffstat (limited to 'modules/textual_inversion/preprocess.py')
-rw-r--r--modules/textual_inversion/preprocess.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/modules/textual_inversion/preprocess.py b/modules/textual_inversion/preprocess.py
index a009d8e8..0d4c3f84 100644
--- a/modules/textual_inversion/preprocess.py
+++ b/modules/textual_inversion/preprocess.py
@@ -47,7 +47,7 @@ def save_pic_with_caption(image, index, params: PreprocessParams, existing_capti
caption += shared.interrogator.generate_caption(image)
if params.process_caption_deepbooru:
- if len(caption) > 0:
+ if caption:
caption += ", "
caption += deepbooru.model.tag_multi(image)
@@ -67,7 +67,7 @@ def save_pic_with_caption(image, index, params: PreprocessParams, existing_capti
caption = caption.strip()
- if len(caption) > 0:
+ if caption:
with open(os.path.join(params.dstdir, f"{basename}.txt"), "w", encoding="utf8") as file:
file.write(caption)