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,6 @@
+AbsArguments
+============
+
+.. autoclass:: FlagEmbedding.abc.finetune.embedder.AbsEmbedderModelArguments
+
+.. autoclass:: FlagEmbedding.abc.finetune.embedder.AbsEmbedderDataArguments
@@ -0,0 +1,32 @@
+==========
+AbsDataset
+==========
+
+AbsRerankerTrainDataset
+=======================
+
+.. autoclass:: FlagEmbedding.abc.finetune.reranker.AbsRerankerTrainDataset
+
+Methods
+-------
+
+.. automethod:: FlagEmbedding.abc.finetune.reranker.AbsRerankerTrainDataset.create_one_example
+
+.. automethod:: FlagEmbedding.abc.finetune.reranker.AbsRerankerTrainDataset._load_dataset
+
+.. automethod:: FlagEmbedding.abc.finetune.reranker.AbsRerankerTrainDataset._shuffle_text
+
+AbsRerankerCollator
+===================
+
+.. autoclass:: FlagEmbedding.abc.finetune.reranker.AbsRerankerCollator
+
+AbsLLMRerankerTrainDataset
+==========================
+
+.. autoclass:: FlagEmbedding.abc.finetune.reranker.AbsLLMRerankerTrainDataset
+
+AbsLLMRerankerCollator
+======================
+
+.. autoclass:: FlagEmbedding.abc.finetune.reranker.AbsLLMRerankerCollator
@@ -0,0 +1,31 @@
+===========
+AbsModeling
+===========
+
+AbsRerankerModel
+================
+
+.. autoclass:: FlagEmbedding.abc.finetune.reranker.AbsRerankerModel
+
+Methods
+-------
+
+.. automethod:: FlagEmbedding.abc.finetune.reranker.AbsRerankerModel.encode
+
+.. automethod:: FlagEmbedding.abc.finetune.reranker.AbsRerankerModel.gradient_checkpointing_enable
+
+.. automethod:: FlagEmbedding.abc.finetune.reranker.AbsRerankerModel.enable_input_require_grads
+
+.. automethod:: FlagEmbedding.abc.finetune.reranker.AbsRerankerModel.forward
+
+.. automethod:: FlagEmbedding.abc.finetune.reranker.AbsRerankerModel.compute_loss
+
+.. automethod:: FlagEmbedding.abc.finetune.reranker.AbsRerankerModel.save
+
+.. automethod:: FlagEmbedding.abc.finetune.reranker.AbsRerankerModel.save_pretrained
+
+
+RerankerOutput
+==============
+
+.. autoclass:: FlagEmbedding.abc.finetune.reranker.RerankerOutput
@@ -0,0 +1,21 @@
+=========
+AbsRunner
+=========
+
+AbsRerankerTrainer
+==================
+
+.. autoclass:: FlagEmbedding.abc.finetune.reranker.AbsRerankerRunner
+
+Methods
+-------
+
+.. automethod:: FlagEmbedding.abc.finetune.reranker.AbsRerankerRunner.load_tokenizer_and_model
+
+.. automethod:: FlagEmbedding.abc.finetune.reranker.AbsRerankerRunner.load_trainer
+
+.. automethod:: FlagEmbedding.abc.finetune.reranker.AbsRerankerRunner.load_train_dataset
+
+.. automethod:: FlagEmbedding.abc.finetune.reranker.AbsRerankerRunner.load_data_collator
+
+.. automethod:: FlagEmbedding.abc.finetune.reranker.AbsRerankerRunner.run
@@ -0,0 +1,13 @@
+==========
+AbsTrainer
+==========
+
+AbsRerankerTrainer
+==================
+
+.. autoclass:: FlagEmbedding.abc.finetune.reranker.AbsRerankerTrainer
+
+Methods
+-------
+
+.. automethod:: FlagEmbedding.abc.finetune.reranker.AbsRerankerTrainer.compute_loss