apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix conversion context

This commit is contained in:
Sean Sube 2023-05-14 20:05:49 -05:00
parent 0d51d61728
commit 56fa0729b5
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
api/onnx_web/convert/utils.py
View File

@ -65,6 +65,7 @@ class ConversionContext(ServerContext):
context = super().from_environ() context = super().from_environ()
context.control = get_boolean(environ, "ONNX_WEB_CONVERT_CONTROL", True) context.control = get_boolean(environ, "ONNX_WEB_CONVERT_CONTROL", True)
context.opset = int(environ.get("ONNX_WEB_CONVERT_OPSET", DEFAULT_OPSET)) context.opset = int(environ.get("ONNX_WEB_CONVERT_OPSET", DEFAULT_OPSET))
return context
def download_progress(urls: List[Tuple[str, str]]): def download_progress(urls: List[Tuple[str, str]]):