Update src\models\moe_vit.py

src/models/moe_vit.py

@@ -413,7 +413,7 @@ class MoEViT(nn.Module):
         self, x: torch.Tensor, pool: str = "patchmean"
     ) -> torch.Tensor:
         """Per-slice features (no head): [S, D] from [S, 3, H, W] slices."""
-        tokens = self.stem.forward_features(x).bfloat16()  # [S, T, D] (bit-equal to feat cache)
+        tokens = self.stem.forward_features(x)  # [S, T, D]
         return self.features_from_tokens(tokens, pool)
 
     def features_from_tokens(