Skip to content

Commit 055d955

Browse files
authored
Fix failing CogVideoX LoRA fuse test (#10352)
fix
1 parent 71cc201 commit 055d955

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/models/embeddings.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -748,10 +748,10 @@ def forward(self, text_embeds: torch.Tensor, image_embeds: torch.Tensor):
748748
pos_embedding = self._get_positional_embeddings(
749749
height, width, pre_time_compression_frames, device=embeds.device
750750
)
751-
pos_embedding = pos_embedding.to(dtype=embeds.dtype)
752751
else:
753752
pos_embedding = self.pos_embedding
754753

754+
pos_embedding = pos_embedding.to(dtype=embeds.dtype)
755755
embeds = embeds + pos_embedding
756756

757757
return embeds

0 commit comments

Comments
 (0)