"""
Model implementations for BGE fine-tuning
"""
from .bge_m3 import BGEM3Model
from .bge_reranker import BGERerankerModel
from .losses import (
    InfoNCELoss,
    M3KnowledgeDistillationLoss,
    ListwiseCrossEntropyLoss,
    PairwiseMarginLoss,
    DynamicListwiseDistillationLoss,
    MultiTaskLoss
)


__all__ = [
    'BGEM3Model',
    'BGERerankerModel',
    'InfoNCELoss',
    'M3KnowledgeDistillationLoss',
    'ListwiseCrossEntropyLoss',
    'PairwiseMarginLoss',
    'DynamicListwiseDistillationLoss',
    'MultiTaskLoss'
]