apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

better error logging during conversion

This commit is contained in:
Sean Sube 2023-02-25 08:22:12 -06:00
parent 9ef89db8b0
commit 86984be5d4
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
api/onnx_web/convert/__main__.py
View File

@ -4,6 +4,7 @@ from logging import getLogger
from os import makedirs, path
from sys import exit
from typing import Any, Dict, List, Optional, Tuple
from traceback import format_exception
from urllib.parse import urlparse
from jsonschema import ValidationError, validate
@ -241,7 +242,7 @@ def convert_models(ctx: ConversionContext, args, models: Models):
)
except Exception as e:
logger.error("error converting diffusion model %s: %s", name, e)
logger.error("error converting diffusion model %s: %s", name, format_exception(type(e), e, e.__traceback__))
if args.upscaling and "upscaling" in models:
for model in models.get("upscaling"):
@ -259,7 +260,7 @@ def convert_models(ctx: ConversionContext, args, models: Models):
)
convert_upscale_resrgan(ctx, model, source)
except Exception as e:
logger.error("error converting upscaling model %s: %s", name, e)
logger.error("error converting upscaling model %s: %s", name, format_exception(type(e), e, e.__traceback__))
if args.correction and "correction" in models:
for model in models.get("correction"):
@ -276,7 +277,7 @@ def convert_models(ctx: ConversionContext, args, models: Models):
)
convert_correction_gfpgan(ctx, model, source)
except Exception as e:
logger.error("error converting correction model %s: %s", name, e)
logger.error("error converting correction model %s: %s", name, format_exception(type(e), e, e.__traceback__))
def main() -> int: