From f92e599c707f5ade6e76d899f6850aec8d013cea Mon Sep 17 00:00:00 2001 From: Yuxuan Zhang <2448370773@qq.com> Date: Tue, 4 Mar 2025 03:42:01 +0800 Subject: [PATCH] Update pipeline_cogview4.py (#10944) --- src/diffusers/pipelines/cogview4/pipeline_cogview4.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/pipelines/cogview4/pipeline_cogview4.py b/src/diffusers/pipelines/cogview4/pipeline_cogview4.py index 097d1b6aed..f2c047fb22 100644 --- a/src/diffusers/pipelines/cogview4/pipeline_cogview4.py +++ b/src/diffusers/pipelines/cogview4/pipeline_cogview4.py @@ -215,7 +215,7 @@ class CogView4Pipeline(DiffusionPipeline): ) text_input_ids = torch.cat([pad_ids, text_input_ids], dim=1) prompt_embeds = self.text_encoder( - text_input_ids.to(self.text_encoder.model.device), output_hidden_states=True + text_input_ids.to(self.text_encoder.device), output_hidden_states=True ).hidden_states[-2] prompt_embeds = prompt_embeds.to(dtype=dtype, device=device)