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

ssube · May 27, 2023 · 3adafea · 3adafea
1 parent 86d458a
commit 3adafea
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/api/onnx_web/convert/utils.py b/api/onnx_web/convert/utils.py
@@ -70,7 +70,7 @@ def __init__(
     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))