chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,7 @@
|
||||
Base
|
||||
====
|
||||
|
||||
.. toctree::
|
||||
base/modeling
|
||||
base/runner
|
||||
base/trainer
|
||||
@@ -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
|
||||
@@ -0,0 +1,5 @@
|
||||
Runner
|
||||
======
|
||||
|
||||
.. autoclass:: FlagEmbedding.finetune.embedder.encoder_only.base.EncoderOnlyEmbedderRunner
|
||||
:members:
|
||||
@@ -0,0 +1,5 @@
|
||||
Trainer
|
||||
=======
|
||||
|
||||
.. autoclass:: FlagEmbedding.finetune.embedder.encoder_only.base.EncoderOnlyEmbedderTrainer
|
||||
:members:
|
||||
@@ -0,0 +1,8 @@
|
||||
M3
|
||||
==
|
||||
|
||||
.. toctree::
|
||||
m3/arguments
|
||||
m3/modeling
|
||||
m3/runner
|
||||
m3/trainer
|
||||
@@ -0,0 +1,6 @@
|
||||
Arguments
|
||||
=========
|
||||
|
||||
.. autoclass:: FlagEmbedding.finetune.embedder.encoder_only.m3.EncoderOnlyEmbedderM3ModelArguments
|
||||
|
||||
.. autoclass:: FlagEmbedding.finetune.embedder.encoder_only.m3.EncoderOnlyEmbedderM3TrainingArguments
|
||||
@@ -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:
|
||||
@@ -0,0 +1,5 @@
|
||||
Runner
|
||||
======
|
||||
|
||||
.. autoclass:: FlagEmbedding.finetune.embedder.encoder_only.m3.EncoderOnlyEmbedderM3Runner
|
||||
:members:
|
||||
@@ -0,0 +1,5 @@
|
||||
Trainer
|
||||
=======
|
||||
|
||||
.. autoclass:: FlagEmbedding.finetune.embedder.encoder_only.m3.EncoderOnlyEmbedderM3Trainer
|
||||
:members:
|
||||
Reference in New Issue
Block a user