File size: 687 Bytes
d1e5d70 998398d d1e5d70 998398d d1e5d70 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
from transformers.models.qwen2_5_vl import Qwen2_5_VLConfig
from typing import Optional
class JinaEmbeddingsV4Config(Qwen2_5_VLConfig):
"""
Configuration for the JinaEmbeddingsV4 model.
"""
def __init__(
self,
single_vector_pool_strategy: str = "mean",
multi_vector_projector_dim: int = 128,
pretrained_peft_model_name_or_path: Optional[str] = None,
**kwargs,
):
super().__init__(**kwargs)
self.single_vector_pool_strategy = single_vector_pool_strategy
self.multi_vector_projector_dim = multi_vector_projector_dim
self.pretrained_peft_model_name_or_path = pretrained_peft_model_name_or_path
|