fix(api): handle HF repos in new SD converter
This commit is contained in:
parent
27bd307cd8
commit
29ccd16437
|
@ -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)
|
||||||
|
|
Loading…
Reference in New Issue