[Serving] fix simple_serving log error (#1117)

* fix simple_serving log error

* 1

1
This commit is contained in:
Thomas Young
2023-01-11 20:12:37 +08:00
committed by GitHub
parent 70bf66bc5d
commit 1feddbcc01
2 changed files with 39 additions and 2 deletions

View File

@@ -132,7 +132,44 @@ def main():
print(
"Model convert failed! Please check if you have installed it!")
if args.tools == "simple_serving":
uvicorn.run(args.app, host=args.host, port=args.port, app_dir='.')
custom_logging_config = {
"version": 1,
"disable_existing_loggers": False,
"formatters": {
"default": {
"()": "uvicorn.logging.DefaultFormatter",
"fmt": "%(asctime)s %(levelprefix)s %(message)s",
'datefmt': '%Y-%m-%d %H:%M:%S',
"use_colors": None,
},
},
"handlers": {
"default": {
"formatter": "default",
"class": "logging.StreamHandler",
"stream": "ext://sys.stderr",
},
'null': {
"formatter": "default",
"class": 'logging.NullHandler'
}
},
"loggers": {
"": {
"handlers": ["null"],
"level": "DEBUG"
},
"uvicorn.error": {
"handlers": ["default"],
"level": "DEBUG"
}
},
}
uvicorn.run(args.app,
host=args.host,
port=args.port,
app_dir='.',
log_config=custom_logging_config)
if __name__ == '__main__':