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

ssube · Dec 15, 2023 · 95d8f4a · 95d8f4a
1 parent 2a641b1
commit 95d8f4a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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(