get_1d_sincos_pos_embed_from_grid in LatteTransformer3DModel

Author: hlky

src/diffusers/models/transformers/latte_transformer_3d.py

@@ -158,7 +158,7 @@ def __init__(
         temp_pos_embed = get_1d_sincos_pos_embed_from_grid(
             inner_dim, torch.arange(0, video_length).unsqueeze(1)
         )  # 1152 hidden size
-        self.register_buffer("temp_pos_embed", torch.from_numpy(temp_pos_embed).float().unsqueeze(0), persistent=False)
+        self.register_buffer("temp_pos_embed", temp_pos_embed.float().unsqueeze(0), persistent=False)
 
         self.gradient_checkpointing = False