From 95d8f4a5986f27255828481ff0b7c5f485584ce3 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Thu, 14 Dec 2023 20:04:08 -0600 Subject: [PATCH] fix(api): make sure all file types are covered (#432) --- api/onnx_web/convert/diffusion/diffusion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/api/onnx_web/convert/diffusion/diffusion.py b/api/onnx_web/convert/diffusion/diffusion.py index 296f7351..da16a5e9 100644 --- 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(