Update modeling_marqo_arctic_bge_chimera_m.py

modeling_marqo_arctic_bge_chimera_m.py

@@ -1,5 +1,6 @@
 import torch
 import torch.nn as nn
+import torch.nn.functional as F
 from transformers import BertModel, PreTrainedModel, BertConfig, AutoModel
 from typing import List
 from .configuration_marqo_arctic_bge_chimera_m import ChimeraConfig
@@ -46,6 +47,7 @@ class Chimera(PreTrainedModel):
                 token_type_ids=token_type_ids,
             )
             pooled_output = model_output[0][:, 0]
+            pooled_output = F.normalize(pooled_output, p=2, dim=-1)
             embeddings.append(pooled_output)
 
         return torch.cat(embeddings, dim=-1)