update

hao-ai-lab · Dec 31, 2024 · e0234ea · e0234ea
1 parent b2a75a5
commit e0234ea
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fastvideo/models/mochi_hf/modeling_mochi.py b/fastvideo/models/mochi_hf/modeling_mochi.py
@@ -489,7 +489,7 @@ def _create_rope(self, freqs: torch.Tensor,
         with torch.autocast(freqs.device.type, enabled=False):
             # Always run ROPE freqs computation in FP32
             freqs = torch.einsum(
-                "nd,dhf->nhf",
+                "nd,dhf->nhf",  # codespell:ignore
                 pos.to(torch.float32),  # codespell:ignore
                 freqs.to(torch.float32))
         freqs_cos = torch.cos(freqs)