fix(api): handle HF repos in new SD converter

ssube · Dec 24, 2023 · 29ccd16 · 29ccd16
1 parent 27bd307
commit 29ccd16
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
@@ -804,7 +804,7 @@ def convert_diffusion_diffusers_optimum(
         cache_path, conversion.map_location, size=image_size, version=version
     )
 
-    if path.isdir(cache_path):
+    if source.startswith(HuggingfaceClient.protocol) or path.isdir(cache_path):
         pipeline = pipe_class.from_pretrained(cache_path, **pipe_args)
     else:
         pipeline = pipe_class.from_single_file(cache_path, **pipe_args)