jupyterjazz commited on
Commit
9aa2833
·
verified ·
1 Parent(s): ef6c560

Update modeling_jina_embeddings_v4.py

Browse files
Files changed (1) hide show
  1. modeling_jina_embeddings_v4.py +1 -0
modeling_jina_embeddings_v4.py CHANGED
@@ -346,6 +346,7 @@ class JinaEmbeddingsV4Model(Qwen2_5_VLForConditionalGeneration):
346
  embeddings = embeddings.single_vec_emb
347
  if truncate_dim is not None:
348
  embeddings = embeddings[:, :truncate_dim]
 
349
  else:
350
  embeddings = embeddings.multi_vec_emb
351
 
 
346
  embeddings = embeddings.single_vec_emb
347
  if truncate_dim is not None:
348
  embeddings = embeddings[:, :truncate_dim]
349
+ embeddings = torch.nn.functional.normalize(embeddings, p=2, dim=-1)
350
  else:
351
  embeddings = embeddings.multi_vec_emb
352