From 666a3d94485ac0aeaacc5f2e4ac5001d858fedc5 Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Mon, 7 Jul 2025 13:47:12 +0530 Subject: [PATCH] update --- src/diffusers/models/embeddings.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/diffusers/models/embeddings.py b/src/diffusers/models/embeddings.py index c24e1d223f..40e559bd75 100644 --- a/src/diffusers/models/embeddings.py +++ b/src/diffusers/models/embeddings.py @@ -1163,8 +1163,8 @@ def get_1d_rotary_pos_embed( if use_real and repeat_interleave_real: # flux, hunyuan-dit, cogvideox # Use dim=-1 for robust interleaving on the feature dimension - freqs_cos = freqs.cos().repeat_interleave(2, dim=-1) # Shape: [B, S, D] - freqs_sin = freqs.sin().repeat_interleave(2, dim=-1) # Shape: [B, S, D] + freqs_cos = freqs.cos().repeat_interleave(2, dim=-1).float() # Shape: [B, S, D] + freqs_sin = freqs.sin().repeat_interleave(2, dim=-1).float() # Shape: [B, S, D] return freqs_cos, freqs_sin elif use_real: # stable audio, allegro