apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): switch back to torch extraction for now (#379)

This commit is contained in:
Sean Sube 2023-05-27 12:29:17 -05:00
parent 86d458a0c6
commit 3adafeaa3d
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -70,7 +70,7 @@ class ConversionContext(ServerContext):
def from_environ(cls):
context = super().from_environ()
context.control = get_boolean(environ, "ONNX_WEB_CONVERT_CONTROL", True)
context.extract = get_boolean(environ, "ONNX_WEB_CONVERT_EXTRACT", False)
context.extract = get_boolean(environ, "ONNX_WEB_CONVERT_EXTRACT", True)
context.reload = get_boolean(environ, "ONNX_WEB_CONVERT_RELOAD", True)
context.share_unet = get_boolean(environ, "ONNX_WEB_CONVERT_SHARE_UNET", True)
context.opset = int(environ.get("ONNX_WEB_CONVERT_OPSET", DEFAULT_OPSET))