diff --git a/scripts/iib/parsers/index.py b/scripts/iib/parsers/index.py index 0f1531a..c5734b9 100644 --- a/scripts/iib/parsers/index.py +++ b/scripts/iib/parsers/index.py @@ -2,6 +2,7 @@ from scripts.iib.parsers.comfyui import ComfyUIParser from scripts.iib.parsers.sd_webui import SdWebUIParser from scripts.iib.parsers.fooocus import FooocusParser from scripts.iib.parsers.model import ImageGenerationInfo +from scripts.iib.logger import logger from PIL import Image @@ -10,5 +11,10 @@ def parse_image_info(image_path: str) -> ImageGenerationInfo: with Image.open(image_path) as img: for parser in parsers: if parser.test(img, image_path): - return parser.parse(img, image_path) + try: + return parser.parse(img, image_path) + except Exception as e: + logger.error(e, stack_info=True) + print(e) + return ImageGenerationInfo() raise Exception("matched parser is not found")