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
|
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
|