devve1 commited on
Commit
27e5aac
1 Parent(s): 9189fb3

Update optimum_encoder.py

Browse files
Files changed (1) hide show
  1. 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)