Skip to content

Commit 72741ec

Browse files
committed
pipeline fixes
1 parent 237e079 commit 72741ec

File tree

2 files changed

+121
-82
lines changed

2 files changed

+121
-82
lines changed

src/diffusers/models/transformers/transformer_mochi.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -205,7 +205,7 @@ def _get_positions(
205205
return positions
206206

207207
def _create_rope(self, freqs: torch.Tensor, pos: torch.Tensor) -> torch.Tensor:
208-
freqs = torch.einsum("nd,dhf->nhf", pos, freqs)
208+
freqs = torch.einsum("nd,dhf->nhf", pos, freqs.float())
209209
freqs_cos = torch.cos(freqs)
210210
freqs_sin = torch.sin(freqs)
211211
return freqs_cos, freqs_sin

0 commit comments

Comments
 (0)