26 lines
536 B
Python
26 lines
536 B
Python
"""
|
|
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'
|
|
]
|