fix(api): convert SD models without extracting Torch intermediate (#404)
This commit is contained in:
parent
1d6547977c
commit
b29837d773
|
@ -69,7 +69,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", True)
|
||||
context.extract = get_boolean(environ, "ONNX_WEB_CONVERT_EXTRACT", False)
|
||||
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))
|
||||
|
|
Loading…
Reference in New Issue