apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

default content type, use arg parser for override

This commit is contained in:
Sean Sube 2023-05-08 21:56:05 -05:00
parent c992c03584
commit 6f38d35c73
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -582,7 +582,7 @@ def main(args = None) -> int:
help="HuggingFace token with read permissions for downloading models.",
)
args = args or parser.parse_args()
args = parser.parse_args(args=args)
logger.info("CLI arguments: %s", args)
server = ConversionContext.from_environ()

5
api/onnx_web/server/admin.py
View File

@ -48,7 +48,7 @@ def update_extra_models(server: ServerContext):
return make_response(jsonify({})), 401
extra_schema = load_config("./schemas/extras.yaml")
data_str = request.data.decode(encoding=request.content_encoding)
data_str = request.data.decode(encoding=(request.content_encoding or "utf-8"))
try:
data = load_config_str(data_str)
@ -63,11 +63,10 @@ def update_extra_models(server: ServerContext):
with open(server.extra_models[0], mode="w") as f:
f.write(data_str)
from argparse import Namespace
from onnx_web.convert.__main__ import main as convert
logger.warning("downloading and converting models to ONNX")
convert(args=Namespace(correction=True, diffusion=True, upscalinng=True, extras=server.extra_models))
convert(args=["--correction", "--diffusion", "--upscaling", "--extras", *server.extra_models])
return jsonify(data)