Update pipeline_cogview4.py (#10944)

zRzRzRzRzRzRzR · web-flow · commit f92e599c707f · 2025-03-03T09:42:01.000-10:00
diff --git 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)

Original file line number	Diff line number	Diff line change
`@@ -215,7 +215,7 @@ def _get_glm_embeds(`
`215`	`215`	`)`
`216`	`216`	`text_input_ids = torch.cat([pad_ids, text_input_ids], dim=1)`
`217`	`217`	`prompt_embeds = self.text_encoder(`
`218`		`- text_input_ids.to(self.text_encoder.model.device), output_hidden_states=True`
	`218`	`+ text_input_ids.to(self.text_encoder.device), output_hidden_states=True`
`219`	`219`	`).hidden_states[-2]`
`220`	`220`
`221`	`221`	`prompt_embeds = prompt_embeds.to(dtype=dtype, device=device)`