update

Author: DN6

src/diffusers/models/transformers/transformer_mochi.py

@@ -451,7 +451,7 @@ def _get_positions(
         return positions
 
     def _create_rope(self, freqs: torch.Tensor, pos: torch.Tensor) -> torch.Tensor:
-        with torch.autocast(self.device.type, torch.float32):
+        with torch.autocast(freqs.device.type, torch.float32):
             # Always run ROPE freqs computation in FP32
             freqs = torch.einsum("nd,dhf->nhf", pos.to(torch.float32), freqs.to(torch.float32))