Update optimum_encoder.py

optimum_encoder.py

@@ -141,7 +141,7 @@ class OptimumEncoder(BaseEncoder):
                 else:
                     embeddings = self._torch.nn.functional.normalize(embeddings, p=2, dim=1).detach().cpu().tolist()
 
-            original_dimensions = embeddings.size(1)
+            original_dimensions = embeddings.size()
 
             if original_dimensions > matryoshka_dim:
                 embeddings = embeddings[:, :matryoshka_dim]