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 os import makedirs, path
from sys import exit from sys import exit
from typing import Any, Dict, List, Optional, Tuple from typing import Any, Dict, List, Optional, Tuple
from traceback import format_exception
from urllib.parse import urlparse from urllib.parse import urlparse
from jsonschema import ValidationError, validate from jsonschema import ValidationError, validate
@ -241,7 +242,7 @@ def convert_models(ctx: ConversionContext, args, models: Models):
) )
except Exception as e: 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: if args.upscaling and "upscaling" in models:
for model in models.get("upscaling"): for model in models.get("upscaling"):
@ -259,7 +260,7 @@ def convert_models(ctx: ConversionContext, args, models: Models):
) )
convert_upscale_resrgan(ctx, model, source) convert_upscale_resrgan(ctx, model, source)
except Exception as e: 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: if args.correction and "correction" in models:
for model in models.get("correction"): for model in models.get("correction"):
@ -276,7 +277,7 @@ def convert_models(ctx: ConversionContext, args, models: Models):
) )
convert_correction_gfpgan(ctx, model, source) convert_correction_gfpgan(ctx, model, source)
except Exception as e: 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: def main() -> int: