1580158
/
sd-webui-infinite-image-browsing
mirror of https://github.com/zanllp/sd-webui-infinite-image-browsing.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

feat: Add exception handling

pull/561/head
zanllp 2024-03-27 01:27:45 +08:00
parent 78e80288f6
commit eb75d4e583
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
scripts/iib/parsers/comfyui.py
View File

@ -21,12 +21,15 @@ class ComfyUIParser:
params = None
if not clz.test(img):
raise Exception("The input image does not match the current parser.")
if is_img_created_by_comfyui_with_webui_gen_info(img):
info = read_sd_webui_gen_info_from_image(img, file_path)
params = parse_generation_parameters(info)
else:
params = get_comfyui_exif_data(img)
info = comfyui_exif_data_to_str(params)
try:
if is_img_created_by_comfyui_with_webui_gen_info(img):
info = read_sd_webui_gen_info_from_image(img, file_path)
params = parse_generation_parameters(info)
else:
params = get_comfyui_exif_data(img)
info = comfyui_exif_data_to_str(params)
except Exception:
return ImageGenerationInfo()
return ImageGenerationInfo(
info,
ImageGenerationParams(