From cc0e7527d78f4ee20c1269420a6a1ea342af260f Mon Sep 17 00:00:00 2001 From: "wenmeng.zwm" Date: Thu, 29 Jun 2023 20:35:57 +0800 Subject: [PATCH] [to #50527737] fix cpu inference for stable diffusion pipeline Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/13120858 --- .../stable_diffusion/stable_diffusion_pipeline.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/modelscope/pipelines/multi_modal/diffusers_wrapped/stable_diffusion/stable_diffusion_pipeline.py b/modelscope/pipelines/multi_modal/diffusers_wrapped/stable_diffusion/stable_diffusion_pipeline.py index c4f8bda8..f09d459d 100644 --- a/modelscope/pipelines/multi_modal/diffusers_wrapped/stable_diffusion/stable_diffusion_pipeline.py +++ b/modelscope/pipelines/multi_modal/diffusers_wrapped/stable_diffusion/stable_diffusion_pipeline.py @@ -34,8 +34,9 @@ class StableDiffusionPipeline(DiffusersPipeline): self.device = 'cuda' if torch.cuda.is_available() else 'cpu' # load pipeline + torch_type = torch.float16 if self.device == 'cuda' else torch.float32 self.pipeline = DiffuserStableDiffusionPipeline.from_pretrained( - model, torch_dtype=torch.float16) + model, torch_dtype=torch_type) self.pipeline = self.pipeline.to(self.device) # load lora moudle to unet if lora_dir is not None: