From a10c830e8cdba71602a9760501301790186f09a2 Mon Sep 17 00:00:00 2001 From: Yuxuan Zhang <2448370773@qq.com> Date: Mon, 3 Mar 2025 19:54:01 +0800 Subject: [PATCH] Update pipeline_cogview4.py --- 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 097d1b6aed41..f2c047fb22c9 100644 --- a/src/diffusers/pipelines/cogview4/pipeline_cogview4.py +++ b/src/diffusers/pipelines/cogview4/pipeline_cogview4.py @@ -215,7 +215,7 @@ def _get_glm_embeds( ) 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)