devve1 commited on
Commit
dfac9fe
1 Parent(s): 7e080fb

Update optimum_encoder.py

Browse files
Files changed (1) hide show
  1. optimum_encoder.py +1 -1
optimum_encoder.py CHANGED
@@ -103,7 +103,7 @@ class OptimumEncoder(BaseEncoder):
103
  normalize_embeddings: bool = True,
104
  pooling_strategy: str = "mean",
105
  matryoshka_dim: int = 1024,
106
- convert_to_numpy: bool = True
107
  ) -> List[List[float]] | List[np.ndarray]:
108
  all_embeddings = []
109
  for i in range(0, len(docs), batch_size):
 
103
  normalize_embeddings: bool = True,
104
  pooling_strategy: str = "mean",
105
  matryoshka_dim: int = 1024,
106
+ convert_to_numpy: bool = False
107
  ) -> List[List[float]] | List[np.ndarray]:
108
  all_embeddings = []
109
  for i in range(0, len(docs), batch_size):