fix(api): correct name of torch extraction var

ssube · Jun 3, 2023 · 4810ca8 · 4810ca8
1 parent a19bf06
commit 4810ca8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/api/onnx_web/convert/diffusion/diffusers.py b/api/onnx_web/convert/diffusion/diffusers.py
@@ -328,7 +328,7 @@ def convert_diffusion_diffusers(
             use_auth_token=conversion.token,
         ).to(device)
     elif path.exists(source) and path.isfile(source):
-        if conversion.extract_torch:
+        if conversion.extract:
             logger.debug("extracting SD checkpoint to Torch models: %s", source)
             torch_source = convert_extract_checkpoint(
                 conversion,