init commit

2025-07-22 16:55:25 +08:00
commit 36003b83e2
67 changed files with 76613 additions and 0 deletions
--- a/models/init.py
+++ b/models/init.py
@@ -0,0 +1,25 @@
+"""
+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'
+]