Spaces:

GameScribes
/

Multipurpose-AI-Agent-Development

Sleeping

devve1 commited on Jul 13

Commit

f14f28b

•

1 Parent(s): 31592b1

Update optimum_encoder.py

Files changed (1) hide show

optimum_encoder.py CHANGED Viewed

@@ -176,7 +176,7 @@ class OptimumEncoder(BaseModel, Embeddings):
         if normalize_embeddings:
             embeddings = self._torch.nn.functional.normalize(embeddings, p=2, dim=1)
         return embeddings.tolist()
     def _mean_pooling(self, model_output, attention_mask):

         if normalize_embeddings:
             embeddings = self._torch.nn.functional.normalize(embeddings, p=2, dim=1)
+        print(embeddings)
         return embeddings.tolist()
     def _mean_pooling(self, model_output, attention_mask):