Mirrors
/
stable-diffusion-webui
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

prevent error spam when processing images without txt files for captions

This commit is contained in:
AUTOMATIC 2022-10-21 18:46:02 +03:00
parent 7464f367c3
commit f49c08ea56
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
modules/textual_inversion/preprocess.py
View File

@ -122,11 +122,10 @@ def preprocess_work(process_src, process_dst, process_width, process_height, pre
continue continue
existing_caption = None existing_caption = None
existing_caption_filename = os.path.splitext(filename)[0] + '.txt'
try: if os.path.exists(existing_caption_filename):
existing_caption = open(os.path.splitext(filename)[0] + '.txt', 'r').read() with open(existing_caption_filename, 'r', encoding="utf8") as file:
except Exception as e: existing_caption = file.read()
print(e)
if shared.state.interrupted: if shared.state.interrupted:
break break