Update optimum_encoder.py
Browse files- optimum_encoder.py +1 -1
optimum_encoder.py
CHANGED
@@ -134,7 +134,7 @@ class OptimumEncoder(BaseEncoder):
|
|
134 |
raise ValueError(
|
135 |
"Invalid pooling_strategy. Please use 'mean' or 'max'."
|
136 |
)
|
137 |
-
|
138 |
if normalize_embeddings:
|
139 |
if convert_to_numpy:
|
140 |
embeddings = normalize(embeddings[:, 0]).astype(np.float32)
|
|
|
134 |
raise ValueError(
|
135 |
"Invalid pooling_strategy. Please use 'mean' or 'max'."
|
136 |
)
|
137 |
+
print(f'Embeddings {embeddings}')
|
138 |
if normalize_embeddings:
|
139 |
if convert_to_numpy:
|
140 |
embeddings = normalize(embeddings[:, 0]).astype(np.float32)
|