Skip to content

Commit d50112c

Browse files
ZiyiXiaZiyiXia
committed
package dependency
1 parent 6f224f1 commit d50112c

2 files changed

Lines changed: 8 additions & 2 deletions

File tree

‎FlagEmbedding/finetune/reranker/decoder_only/base/__init__.py‎

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,10 @@
11
from .modeling import CrossDecoderModel
22
from .runner import DecoderOnlyRerankerRunner
3+
from .arguments import RerankerModelArguments
4+
from .trainer import DecoderOnlyRerankerTrainer
35

46
__all__ = [
57
"CrossDecoderModel",
6-
"DecoderOnlyRerankerRunner"
8+
"DecoderOnlyRerankerRunner",
9+
"DecoderOnlyRerankerTrainer"
710
]

‎FlagEmbedding/finetune/reranker/decoder_only/layerwise/__init__.py‎

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,10 @@
11
from .modeling import CrossDecoderModel
22
from .runner import DecoderOnlyRerankerRunner
3+
from .arguments import RerankerModelArguments
4+
from .trainer import DecoderOnlyRerankerTrainer
35

46
__all__ = [
57
"CrossDecoderModel",
6-
"DecoderOnlyRerankerRunner"
8+
"DecoderOnlyRerankerRunner",
9+
"DecoderOnlyRerankerTrainer"
710
]

0 commit comments

Comments
 (0)