From bfe8b4131505d4ee722276abfe1d752234ffd465 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Thu, 13 Jul 2023 23:16:43 +0200 Subject: [PATCH] [From Single File] Force accelerate to be installed (#4078) force accelerate to be installed --- .../pipelines/stable_diffusion/convert_from_ckpt.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py b/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py index a9094cf12f..599f7826b7 100644 --- 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