wehub/flagopen--flagembedding
1
0
Fork 0
Code Issues Pull Requests Actions Deployments Models Agent Notes Packages Projects Releases Wiki Activity

chore: import upstream snapshot with attribution

Browse Source
This commit is contained in:
wehub-resource-sync
2026-07-13 13:39:21 +08:00
commit bc34f6df14
1149 changed files with 328099 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docs/source/API/finetune/embedder.rst
+6
View File
@@ -0,0 +1,6 @@
Embedder
========
.. toctree::
embedder/encoder_only
embedder/decoder_only
docs/source/API/finetune/embedder/decoder_only.rst
+6
View File
@@ -0,0 +1,6 @@
Decoder Only
============
.. toctree::
decoder_only/base
decoder_only/icl
docs/source/API/finetune/embedder/decoder_only/base.rst
+8
View File
@@ -0,0 +1,8 @@
Base
====
.. toctree::
base/arguments
base/modeling
base/runner
base/trainer
docs/source/API/finetune/embedder/decoder_only/base/arguments.rst
+4
View File
@@ -0,0 +1,4 @@
Arguments
=========
.. autoclass:: FlagEmbedding.finetune.embedder.decoder_only.base.DecoderOnlyEmbedderModelArguments
docs/source/API/finetune/embedder/decoder_only/base/modeling.rst
+10
View File
@@ -0,0 +1,10 @@
========
Modeling
========
.. autoclass:: FlagEmbedding.finetune.reranker.decoder_only.base.CrossDecoderModel
Methods
=======
.. automethod:: FlagEmbedding.finetune.reranker.decoder_only.base.CrossDecoderModel.encode
docs/source/API/finetune/embedder/decoder_only/base/runner.rst
+5
View File
@@ -0,0 +1,5 @@
Runner
======
.. autoclass:: FlagEmbedding.finetune.reranker.decoder_only.base.DecoderOnlyRerankerRunner
:members:
docs/source/API/finetune/embedder/decoder_only/base/trainer.rst
+5
View File
@@ -0,0 +1,5 @@
Trainer
=======
.. autoclass:: FlagEmbedding.finetune.reranker.decoder_only.base.DecoderOnlyRerankerTrainer
:members:
docs/source/API/finetune/embedder/decoder_only/icl.rst
+9
View File
@@ -0,0 +1,9 @@
ICL
===
.. toctree::
icl/arguments
icl/dataset
icl/modeling
icl/runner
icl/trainer
docs/source/API/finetune/embedder/decoder_only/icl/arguments.rst
+6
View File
@@ -0,0 +1,6 @@
Arguments
=========
.. autoclass:: FlagEmbedding.finetune.embedder.decoder_only.icl.DecoderOnlyEmbedderICLModelArguments
.. autoclass:: FlagEmbedding.finetune.embedder.decoder_only.icl.DecoderOnlyEmbedderICLDataArguments
docs/source/API/finetune/embedder/decoder_only/icl/dataset.rst
+18
View File
@@ -0,0 +1,18 @@
=======
Dataset
=======
DecoderOnlyEmbedderICLSameDatasetTrainDataset
=============================================
.. autoclass:: FlagEmbedding.finetune.embedder.decoder_only.icl.DecoderOnlyEmbedderICLSameDatasetTrainDataset
Methods
-------
.. automethod:: FlagEmbedding.finetune.embedder.decoder_only.icl.DecoderOnlyEmbedderICLSameDatasetTrainDataset._create_batch_data
AbsEmbedderSameDatasetCollator
==============================
.. autoclass:: FlagEmbedding.finetune.embedder.decoder_only.icl.AbsEmbedderSameDatasetCollator
docs/source/API/finetune/embedder/decoder_only/icl/modeling.rst
+18
View File
@@ -0,0 +1,18 @@
========
Modeling
========
.. autoclass:: FlagEmbedding.finetune.embedder.decoder_only.icl.BiDecoderOnlyEmbedderICLModel
Methods
=======
.. automethod:: FlagEmbedding.finetune.embedder.decoder_only.icl.BiDecoderOnlyEmbedderICLModel.encode
.. automethod:: FlagEmbedding.finetune.embedder.decoder_only.icl.BiDecoderOnlyEmbedderICLModel.compute_score
.. automethod:: FlagEmbedding.finetune.embedder.decoder_only.icl.BiDecoderOnlyEmbedderICLModel.compute_loss
.. automethod:: FlagEmbedding.finetune.embedder.decoder_only.icl.BiDecoderOnlyEmbedderICLModel.gradient_checkpointing_enable
.. automethod:: FlagEmbedding.finetune.embedder.decoder_only.icl.BiDecoderOnlyEmbedderICLModel.enable_input_require_grads
.. automethod:: FlagEmbedding.finetune.embedder.decoder_only.icl.BiDecoderOnlyEmbedderICLModel.save
.. automethod:: FlagEmbedding.finetune.embedder.decoder_only.icl.BiDecoderOnlyEmbedderICLModel._sentence_embedding
.. automethod:: FlagEmbedding.finetune.embedder.decoder_only.icl.BiDecoderOnlyEmbedderICLModel._compute_similarity
docs/source/API/finetune/embedder/decoder_only/icl/runner.rst
+5
View File
@@ -0,0 +1,5 @@
Runner
======
.. autoclass:: FlagEmbedding.finetune.embedder.decoder_only.icl.DecoderOnlyEmbedderICLRunner
:members:
docs/source/API/finetune/embedder/decoder_only/icl/trainer.rst
+5
View File
@@ -0,0 +1,5 @@
Trainer
=======
.. autoclass:: FlagEmbedding.finetune.embedder.decoder_only.icl.DecoderOnlyEmbedderICLTrainer
:members:
docs/source/API/finetune/embedder/encoder_only.rst
+6
View File
@@ -0,0 +1,6 @@
Encoder Only
============
.. toctree::
encoder_only/base
encoder_only/m3
docs/source/API/finetune/embedder/encoder_only/base.rst
+7
View File
@@ -0,0 +1,7 @@
Base
====
.. toctree::
base/modeling
base/runner
base/trainer
docs/source/API/finetune/embedder/encoder_only/base/modeling.rst
+17
View File
@@ -0,0 +1,17 @@
Modeling
========
.. autoclass:: FlagEmbedding.finetune.embedder.encoder_only.base.BiEncoderOnlyEmbedderModel
Methods
-------
.. automethod:: FlagEmbedding.finetune.embedder.encoder_only.base.BiEncoderOnlyEmbedderModel.encode
.. automethod:: FlagEmbedding.finetune.embedder.encoder_only.base.BiEncoderOnlyEmbedderModel.compute_score
.. automethod:: FlagEmbedding.finetune.embedder.encoder_only.base.BiEncoderOnlyEmbedderModel.compute_loss
.. automethod:: FlagEmbedding.finetune.embedder.encoder_only.base.BiEncoderOnlyEmbedderModel.gradient_checkpointing_enable
.. automethod:: FlagEmbedding.finetune.embedder.encoder_only.base.BiEncoderOnlyEmbedderModel.enable_input_require_grads
.. automethod:: FlagEmbedding.finetune.embedder.encoder_only.base.BiEncoderOnlyEmbedderModel.save
.. automethod:: FlagEmbedding.finetune.embedder.encoder_only.base.BiEncoderOnlyEmbedderModel._sentence_embedding
.. automethod:: FlagEmbedding.finetune.embedder.encoder_only.base.BiEncoderOnlyEmbedderModel._compute_similarity
docs/source/API/finetune/embedder/encoder_only/base/runner.rst
+5
View File
@@ -0,0 +1,5 @@
Runner
======
.. autoclass:: FlagEmbedding.finetune.embedder.encoder_only.base.EncoderOnlyEmbedderRunner
:members:
docs/source/API/finetune/embedder/encoder_only/base/trainer.rst
+5
View File
@@ -0,0 +1,5 @@
Trainer
=======
.. autoclass:: FlagEmbedding.finetune.embedder.encoder_only.base.EncoderOnlyEmbedderTrainer
:members:
docs/source/API/finetune/embedder/encoder_only/m3.rst
+8
View File
@@ -0,0 +1,8 @@
M3
==
.. toctree::
m3/arguments
m3/modeling
m3/runner
m3/trainer
docs/source/API/finetune/embedder/encoder_only/m3/arguments.rst
+6
View File
@@ -0,0 +1,6 @@
Arguments
=========
.. autoclass:: FlagEmbedding.finetune.embedder.encoder_only.m3.EncoderOnlyEmbedderM3ModelArguments
.. autoclass:: FlagEmbedding.finetune.embedder.encoder_only.m3.EncoderOnlyEmbedderM3TrainingArguments
docs/source/API/finetune/embedder/encoder_only/m3/modeling.rst
+35
View File
@@ -0,0 +1,35 @@
========
Modeling
========
EncoderOnlyEmbedderM3Model
============================
.. autoclass:: FlagEmbedding.finetune.embedder.encoder_only.m3.EncoderOnlyEmbedderM3Model
Methods
-------
.. automethod:: FlagEmbedding.finetune.embedder.encoder_only.m3.EncoderOnlyEmbedderM3Model.encode
.. automethod:: FlagEmbedding.finetune.embedder.encoder_only.m3.EncoderOnlyEmbedderM3Model.compute_score
.. automethod:: FlagEmbedding.finetune.embedder.encoder_only.m3.EncoderOnlyEmbedderM3Model.compute_dense_score
.. automethod:: FlagEmbedding.finetune.embedder.encoder_only.m3.EncoderOnlyEmbedderM3Model.compute_sparse_score
.. automethod:: FlagEmbedding.finetune.embedder.encoder_only.m3.EncoderOnlyEmbedderM3Model.compute_colbert_score
.. automethod:: FlagEmbedding.finetune.embedder.encoder_only.m3.EncoderOnlyEmbedderM3Model.ensemble_score
.. automethod:: FlagEmbedding.finetune.embedder.encoder_only.m3.EncoderOnlyEmbedderM3Model.forward
.. automethod:: FlagEmbedding.finetune.embedder.encoder_only.m3.EncoderOnlyEmbedderM3Model.compute_loss
.. automethod:: FlagEmbedding.finetune.embedder.encoder_only.m3.EncoderOnlyEmbedderM3Model.gradient_checkpointing_enable
.. automethod:: FlagEmbedding.finetune.embedder.encoder_only.m3.EncoderOnlyEmbedderM3Model.enable_input_require_grads
.. automethod:: FlagEmbedding.finetune.embedder.encoder_only.m3.EncoderOnlyEmbedderM3Model.save
.. automethod:: FlagEmbedding.finetune.embedder.encoder_only.m3.EncoderOnlyEmbedderM3Model._dense_embedding
.. automethod:: FlagEmbedding.finetune.embedder.encoder_only.m3.EncoderOnlyEmbedderM3Model._sparse_embedding
.. automethod:: FlagEmbedding.finetune.embedder.encoder_only.m3.EncoderOnlyEmbedderM3Model._colbert_embedding
.. automethod:: FlagEmbedding.finetune.embedder.encoder_only.m3.EncoderOnlyEmbedderM3Model._encode
.. automethod:: FlagEmbedding.finetune.embedder.encoder_only.m3.EncoderOnlyEmbedderM3Model._compute_similarity
.. automethod:: FlagEmbedding.finetune.embedder.encoder_only.m3.EncoderOnlyEmbedderM3Model._get_queries_attention_mask
EncoderOnlyEmbedderM3ModelForInference
======================================
.. autoclass:: FlagEmbedding.finetune.embedder.encoder_only.m3.EncoderOnlyEmbedderM3ModelForInference
:members:
docs/source/API/finetune/embedder/encoder_only/m3/runner.rst
+5
View File
@@ -0,0 +1,5 @@
Runner
======
.. autoclass:: FlagEmbedding.finetune.embedder.encoder_only.m3.EncoderOnlyEmbedderM3Runner
:members:
docs/source/API/finetune/embedder/encoder_only/m3/trainer.rst
+5
View File
@@ -0,0 +1,5 @@
Trainer
=======
.. autoclass:: FlagEmbedding.finetune.embedder.encoder_only.m3.EncoderOnlyEmbedderM3Trainer
:members:
docs/source/API/finetune/reranker.rst
+6
View File
@@ -0,0 +1,6 @@
Reranker
========
.. toctree::
reranker/encoder_only
reranker/decoder_only
docs/source/API/finetune/reranker/decoder_only.rst
+6
View File
@@ -0,0 +1,6 @@
Decoder Only
============
.. toctree::
decoder_only/base
decoder_only/layerwise
docs/source/API/finetune/reranker/decoder_only/base.rst
+8
View File
@@ -0,0 +1,8 @@
Base
====
.. toctree::
base/arguments
base/modeling
base/runner
base/trainer
docs/source/API/finetune/reranker/decoder_only/base/arguments.rst
+4
View File
@@ -0,0 +1,4 @@
Arguments
=========
.. autoclass:: FlagEmbedding.finetune.reranker.decoder_only.base.RerankerModelArguments
docs/source/API/finetune/reranker/decoder_only/base/modeling.rst
+18
View File
@@ -0,0 +1,18 @@
========
Modeling
========
.. autoclass:: FlagEmbedding.finetune.embedder.decoder_only.base.BiDecoderOnlyEmbedderModel
Methods
=======
.. automethod:: FlagEmbedding.finetune.embedder.decoder_only.base.BiDecoderOnlyEmbedderModel.encode
.. automethod:: FlagEmbedding.finetune.embedder.decoder_only.base.BiDecoderOnlyEmbedderModel.compute_score
.. automethod:: FlagEmbedding.finetune.embedder.decoder_only.base.BiDecoderOnlyEmbedderModel.compute_loss
.. automethod:: FlagEmbedding.finetune.embedder.decoder_only.base.BiDecoderOnlyEmbedderModel.gradient_checkpointing_enable
.. automethod:: FlagEmbedding.finetune.embedder.decoder_only.base.BiDecoderOnlyEmbedderModel.enable_input_require_grads
.. automethod:: FlagEmbedding.finetune.embedder.decoder_only.base.BiDecoderOnlyEmbedderModel.save
.. automethod:: FlagEmbedding.finetune.embedder.decoder_only.base.BiDecoderOnlyEmbedderModel._sentence_embedding
.. automethod:: FlagEmbedding.finetune.embedder.decoder_only.base.BiDecoderOnlyEmbedderModel._compute_similarity
docs/source/API/finetune/reranker/decoder_only/base/runner.rst
+5
View File
@@ -0,0 +1,5 @@
Runner
======
.. autoclass:: FlagEmbedding.finetune.embedder.decoder_only.base.DecoderOnlyEmbedderRunner
:members:
docs/source/API/finetune/reranker/decoder_only/base/trainer.rst
+5
View File
@@ -0,0 +1,5 @@
Trainer
=======
.. autoclass:: FlagEmbedding.finetune.embedder.decoder_only.base.DecoderOnlyEmbedderTrainer
:members:
docs/source/API/finetune/reranker/decoder_only/layerwise.rst
+8
View File
@@ -0,0 +1,8 @@
Layerwise
=========
.. toctree::
layerwise/arguments
layerwise/modeling
layerwise/runner
layerwise/trainer
docs/source/API/finetune/reranker/decoder_only/layerwise/arguments.rst
+4
View File
@@ -0,0 +1,4 @@
Arguments
=========
.. autoclass:: FlagEmbedding.finetune.reranker.decoder_only.layerwise.RerankerModelArguments
docs/source/API/finetune/reranker/decoder_only/layerwise/modeling.rst
+11
View File
@@ -0,0 +1,11 @@
========
Modeling
========
.. autoclass:: FlagEmbedding.finetune.reranker.decoder_only.layerwise.CrossDecoderModel
Methods
=======
.. automethod:: FlagEmbedding.finetune.reranker.decoder_only.layerwise.CrossDecoderModel.encode
.. automethod:: FlagEmbedding.finetune.reranker.decoder_only.layerwise.CrossDecoderModel.forward
docs/source/API/finetune/reranker/decoder_only/layerwise/runner.rst
+5
View File
@@ -0,0 +1,5 @@
Runner
======
.. autoclass:: FlagEmbedding.finetune.reranker.decoder_only.layerwise.DecoderOnlyRerankerRunner
:members:
docs/source/API/finetune/reranker/decoder_only/layerwise/trainer.rst
+5
View File
@@ -0,0 +1,5 @@
Trainer
=======
.. autoclass:: FlagEmbedding.finetune.reranker.decoder_only.layerwise.DecoderOnlyRerankerTrainer
:members:
docs/source/API/finetune/reranker/encoder_only.rst
+5
View File
@@ -0,0 +1,5 @@
Encoder Only
============
.. toctree::
encoder_only/base
docs/source/API/finetune/reranker/encoder_only/base.rst
+7
View File
@@ -0,0 +1,7 @@
Base
====
.. toctree::
base/modeling
base/runner
base/trainer
docs/source/API/finetune/reranker/encoder_only/base/modeling.rst
+9
View File
@@ -0,0 +1,9 @@
Modeling
========
.. autoclass:: FlagEmbedding.finetune.reranker.encoder_only.base.CrossEncoderModel
Methods
-------
.. automethod:: FlagEmbedding.finetune.reranker.encoder_only.base.CrossEncoderModel.encode
docs/source/API/finetune/reranker/encoder_only/base/runner.rst
+5
View File
@@ -0,0 +1,5 @@
Runner
======
.. autoclass:: FlagEmbedding.finetune.reranker.encoder_only.base.EncoderOnlyRerankerRunner
:members:
docs/source/API/finetune/reranker/encoder_only/base/trainer.rst
+5
View File
@@ -0,0 +1,5 @@
Trainer
=======
.. autoclass:: FlagEmbedding.finetune.reranker.encoder_only.base.EncoderOnlyRerankerTrainer
:members: