Spaces:
Running
on
T4
Running
on
T4
Update optimum_encoder.py
Browse files- 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 =
|
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):
|