fix(api): make sure all file types are covered (#432)

2023-12-14 20:04:08 -06:00 · 2023-12-14 20:04:08 -06:00 · 95d8f4a598
parent 2a641b111e
commit 95d8f4a598
1 changed files with 1 additions and 1 deletions
--- a/api/onnx_web/convert/diffusion/diffusion.py
+++ b/api/onnx_web/convert/diffusion/diffusion.py
@ -346,7 +346,7 @@ def convert_diffusion_diffusers(
                torch_dtype=dtype,
                use_auth_token=conversion.token,
            ).to(device)
-        elif path.isfile(source):
+        else:
            if conversion.extract:
                logger.debug("extracting SD checkpoint to Torch models: %s", source)
                torch_source = convert_extract_checkpoint(