chore: import upstream snapshot with attribution

2026-07-13 13:39:21 +08:00
commit bc34f6df14
1149 changed files with 328099 additions and 0 deletions
@@ -0,0 +1,8 @@
+Base
+====
+
+.. toctree::
+    base/arguments
+    base/modeling
+    base/runner
+    base/trainer
@@ -0,0 +1,4 @@
+Arguments
+=========
+
+.. autoclass:: FlagEmbedding.finetune.embedder.decoder_only.base.DecoderOnlyEmbedderModelArguments
@@ -0,0 +1,10 @@
+========
+Modeling
+========
+
+.. autoclass:: FlagEmbedding.finetune.reranker.decoder_only.base.CrossDecoderModel
+
+Methods
+=======
+
+.. automethod:: FlagEmbedding.finetune.reranker.decoder_only.base.CrossDecoderModel.encode
@@ -0,0 +1,5 @@
+Runner
+======
+
+.. autoclass:: FlagEmbedding.finetune.reranker.decoder_only.base.DecoderOnlyRerankerRunner
+    :members:
@@ -0,0 +1,5 @@
+Trainer
+=======
+
+.. autoclass:: FlagEmbedding.finetune.reranker.decoder_only.base.DecoderOnlyRerankerTrainer
+    :members:
@@ -0,0 +1,9 @@
+ICL
+===
+
+.. toctree::
+    icl/arguments
+    icl/dataset
+    icl/modeling
+    icl/runner
+    icl/trainer
@@ -0,0 +1,6 @@
+Arguments
+=========
+
+.. autoclass:: FlagEmbedding.finetune.embedder.decoder_only.icl.DecoderOnlyEmbedderICLModelArguments
+
+.. autoclass:: FlagEmbedding.finetune.embedder.decoder_only.icl.DecoderOnlyEmbedderICLDataArguments
@@ -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
@@ -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
@@ -0,0 +1,5 @@
+Runner
+======
+
+.. autoclass:: FlagEmbedding.finetune.embedder.decoder_only.icl.DecoderOnlyEmbedderICLRunner
+    :members:
@@ -0,0 +1,5 @@
+Trainer
+=======
+
+.. autoclass:: FlagEmbedding.finetune.embedder.decoder_only.icl.DecoderOnlyEmbedderICLTrainer
+    :members: