Skip to content

GitLab

Unverified Commit f92e599c authored by Yuxuan Zhang's avatar Yuxuan Zhang Committed by GitHub
Browse files

Update pipeline_cogview4.py (#10944)

parent 982f9b38
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/diffusers/pipelines/cogview4/pipeline_cogview4.py
View file @ f92e599c
...@@ -215,7 +215,7 @@ class CogView4Pipeline(DiffusionPipeline): ...@@ -215,7 +215,7 @@ class CogView4Pipeline(DiffusionPipeline):
) )
text_input_ids = torch.cat([pad_ids, text_input_ids], dim=1) text_input_ids = torch.cat([pad_ids, text_input_ids], dim=1)
prompt_embeds = self.text_encoder( 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] ).hidden_states[-2]
prompt_embeds = prompt_embeds.to(dtype=dtype, device=device) prompt_embeds = prompt_embeds.to(dtype=dtype, device=device)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment