AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

[From Single File] Force accelerate to be installed (#4078)

Browse Source 
force accelerate to be installed
This commit is contained in:
Patrick von Platen
2023-07-13 23:16:43 +02:00
committed by GitHub
parent 4535088cec
commit bfe8b41315
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py
View File

@@ -1172,6 +1172,11 @@ def download_from_original_stable_diffusion_ckpt(
StableUnCLIPPipeline,
)
if not is_accelerate_available():
raise ImportError(
"To correctly use `from_single_file`, please make sure that `accelerate` is installed. You can install it with `pip install accelerate`."
)
if pipeline_class is None:
pipeline_class = StableDiffusionPipeline