From 8debc432cf85d8f72b171908e914a019cc1e6706 Mon Sep 17 00:00:00 2001 From: "guangpan.cd" Date: Wed, 19 Apr 2023 17:11:25 +0800 Subject: [PATCH] Quick fix to ChineseStableDiffusion meta tensor bug. Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/12382517 * Quick fix to ChineseStableDiffusion meta tensor bug. --- .../stable_diffusion/chinese_stable_diffusion_pipeline.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/modelscope/pipelines/multi_modal/diffusers_wrapped/stable_diffusion/chinese_stable_diffusion_pipeline.py b/modelscope/pipelines/multi_modal/diffusers_wrapped/stable_diffusion/chinese_stable_diffusion_pipeline.py index d1627962..231b4e75 100644 --- a/modelscope/pipelines/multi_modal/diffusers_wrapped/stable_diffusion/chinese_stable_diffusion_pipeline.py +++ b/modelscope/pipelines/multi_modal/diffusers_wrapped/stable_diffusion/chinese_stable_diffusion_pipeline.py @@ -46,7 +46,9 @@ class ChineseStableDiffusionPipeline(DiffusersPipeline): torch_dtype = kwargs.get('torch_dtype', torch.float32) self.pipeline = _DiffuersChineseStableDiffusionPipeline.from_pretrained( - model, torch_dtype=torch_dtype).to(self.device) + model, torch_dtype=torch_dtype) + self.pipeline.text_encoder.pooler = None + self.pipeline.to(self.device) def forward(self, inputs: Dict[str, Any], **forward_params) -> Dict[str, Any]: