Skip to content

Commit aa9a4a9

Browse files
committed
add dataset classes
1 parent 0e56bd9 commit aa9a4a9

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

  • FlagEmbedding/finetune/embedder/decoder_only/icl

FlagEmbedding/finetune/embedder/decoder_only/icl/__init__.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,10 @@
66
DecoderOnlyEmbedderICLModelArguments,
77
DecoderOnlyEmbedderICLDataArguments
88
)
9+
from .dataset import (
10+
DecoderOnlyEmbedderICLSameDatasetTrainDataset,
11+
AbsEmbedderSameDatasetCollator
12+
)
913
from .modeling import BiDecoderOnlyEmbedderICLModel
1014
from .trainer import DecoderOnlyEmbedderICLTrainer
1115
from .runner import DecoderOnlyEmbedderICLRunner

0 commit comments

Comments
 (0)