From 29ccd1643727935c8e4405531e0c3a6b23a4ac95 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sun, 24 Dec 2023 07:05:56 -0600 Subject: [PATCH] fix(api): handle HF repos in new SD converter --- 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 625e61cb..5ecb725a 100644 --- 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)