apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

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

This commit is contained in:
Sean Sube 2023-12-24 07:05:56 -06:00
parent 27bd307cd8
commit 29ccd16437
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/onnx_web/convert/diffusion/diffusion.py
View File

@ -804,7 +804,7 @@ def convert_diffusion_diffusers_optimum(
cache_path, conversion.map_location, size=image_size, version=version 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) pipeline = pipe_class.from_pretrained(cache_path, **pipe_args)
else: else:
pipeline = pipe_class.from_single_file(cache_path, **pipe_args) pipeline = pipe_class.from_single_file(cache_path, **pipe_args)