dariakryvosheieva commited on
Commit
2d71296
·
verified ·
1 Parent(s): 7190a78

Rename modeling_jina_embeddings_c1.py to modeling_jina_code_embeddings.py

Browse files
modeling_jina_embeddings_c1.py → modeling_jina_code_embeddings.py RENAMED
@@ -50,7 +50,7 @@ def last_token_pooling(model_output, attention_mask):
50
  return token_embeddings[torch.arange(batch_size, device=token_embeddings.device), sequence_lengths].float()
51
 
52
 
53
- class JinaEmbeddingsC1Model(Qwen2Model):
54
  def __init__(self, config: Qwen2Config):
55
  Qwen2Model.__init__(self, config)
56
  self.instructions = INSTRUCTION_CONFIG
 
50
  return token_embeddings[torch.arange(batch_size, device=token_embeddings.device), sequence_lengths].float()
51
 
52
 
53
+ class JinaCodeEmbeddingsModel(Qwen2Model):
54
  def __init__(self, config: Qwen2Config):
55
  Qwen2Model.__init__(self, config)
56
  self.instructions = INSTRUCTION_CONFIG