yujunjun/FlagEmbedding
1
0
Fork 0
mirror of https://github.com/FlagOpen/FlagEmbedding.git synced 2026-01-08 05:03:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

add dataset classes

Browse Source
This commit is contained in:
ZiyiXia 2024-11-07 09:14:41 +00:00
parent 0e56bd9c38
commit aa9a4a9a97
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
FlagEmbedding/finetune/embedder/decoder_only/icl/__init__.py
View File

@ -6,6 +6,10 @@ from .arguments import (
DecoderOnlyEmbedderICLModelArguments,
DecoderOnlyEmbedderICLDataArguments
)
from .dataset import (
DecoderOnlyEmbedderICLSameDatasetTrainDataset,
AbsEmbedderSameDatasetCollator
)
from .modeling import BiDecoderOnlyEmbedderICLModel
from .trainer import DecoderOnlyEmbedderICLTrainer
from .runner import DecoderOnlyEmbedderICLRunner