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

force accelerate to be installed

[From Single File] Force accelerate to be installed (#4078)
force accelerate to be installed
bfe8b413 · Patrick von Platen · GitHub · 4535088c · bfe8b413
Unverified Commit bfe8b413 authored Jul 13, 2023 by Patrick von Platen Committed by GitHub Jul 13, 2023
Show whitespace changes
Inline Side-by-side

Showing with 5 additions and 0 deletions

src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py ...diffusers/pipelines/stable_diffusion/convert_from_ckpt.py +5 -0

No files found.
--- a/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py
+++ b/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py
@@ -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