FluxPosEmbed: Remove Squeeze No-op (huggingface#9409)

Remove Squeeze op

Signed-off-by: Asfiya Baig <[email protected]>
Co-authored-by: YiYi Xu <[email protected]>

Author: asfiyab-nvidia

src/diffusers/models/embeddings.py

@@ -690,7 +690,7 @@ def forward(self, ids: torch.Tensor) -> torch.Tensor:
         n_axes = ids.shape[-1]
         cos_out = []
         sin_out = []
-        pos = ids.squeeze().float()
+        pos = ids.float()
         is_mps = ids.device.type == "mps"
         freqs_dtype = torch.float32 if is_mps else torch.float64
         for i in range(n_axes):