Update src\models\mor_vit.py

src/models/mor_vit.py

@@ -358,7 +358,7 @@ class MoRViT(nn.Module):
         self, x: torch.Tensor, step: int = 0, 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, step, pool)
 
     def features_from_tokens(