aboutsummaryrefslogtreecommitdiffstats
path: root/modules/textual_inversion/preprocess.py
diff options
context:
space:
mode:
authorAarni Koskela <akx@iki.fi>2023-05-09 19:17:58 +0000
committerAarni Koskela <akx@iki.fi>2023-05-09 19:25:39 +0000
commit3ba6c3c83c0983a025c7bddc08bb7f49481b3cbb (patch)
tree77b2bea31c0e94ac67fddf86fdc7cba65a24af2c /modules/textual_inversion/preprocess.py
parent8fb16ceb288a93f6ecaa45a0afcb3db718333e3e (diff)
downloadstable-diffusion-webui-gfx803-3ba6c3c83c0983a025c7bddc08bb7f49481b3cbb.tar.gz
stable-diffusion-webui-gfx803-3ba6c3c83c0983a025c7bddc08bb7f49481b3cbb.tar.bz2
stable-diffusion-webui-gfx803-3ba6c3c83c0983a025c7bddc08bb7f49481b3cbb.zip
Fix up string formatting/concatenation to f-strings where feasible
Diffstat (limited to 'modules/textual_inversion/preprocess.py')
-rw-r--r--modules/textual_inversion/preprocess.py6
1 files changed, 3 insertions, 3 deletions
diff --git a/modules/textual_inversion/preprocess.py b/modules/textual_inversion/preprocess.py
index 4a29151d..da0bcb26 100644
--- a/modules/textual_inversion/preprocess.py
+++ b/modules/textual_inversion/preprocess.py
@@ -63,9 +63,9 @@ def save_pic_with_caption(image, index, params: PreprocessParams, existing_capti
image.save(os.path.join(params.dstdir, f"{basename}.png"))
if params.preprocess_txt_action == 'prepend' and existing_caption:
- caption = existing_caption + ' ' + caption
+ caption = f"{existing_caption} {caption}"
elif params.preprocess_txt_action == 'append' and existing_caption:
- caption = caption + ' ' + existing_caption
+ caption = f"{caption} {existing_caption}"
elif params.preprocess_txt_action == 'copy' and existing_caption:
caption = existing_caption
@@ -174,7 +174,7 @@ def preprocess_work(process_src, process_dst, process_width, process_height, pre
params.src = filename
existing_caption = None
- existing_caption_filename = os.path.splitext(filename)[0] + '.txt'
+ existing_caption_filename = f"{os.path.splitext(filename)[0]}.txt"
if os.path.exists(existing_caption_filename):
with open(existing_caption_filename, 'r', encoding="utf8") as file:
existing_caption = file.read()