Embedder

• Encoder Only
  • Base
    • Modeling
      • BiEncoderOnlyEmbedderModel
      • Methods
        • BiEncoderOnlyEmbedderModel.encode()
        • BiEncoderOnlyEmbedderModel.compute_score()
        • BiEncoderOnlyEmbedderModel.compute_loss()
        • BiEncoderOnlyEmbedderModel.gradient_checkpointing_enable()
        • BiEncoderOnlyEmbedderModel.enable_input_require_grads()
        • BiEncoderOnlyEmbedderModel.save()
        • BiEncoderOnlyEmbedderModel._sentence_embedding()
        • BiEncoderOnlyEmbedderModel._compute_similarity()
    • Runner
      • EncoderOnlyEmbedderRunner
        • EncoderOnlyEmbedderRunner.load_tokenizer_and_model()
        • EncoderOnlyEmbedderRunner.load_trainer()
    • Trainer
      • EncoderOnlyEmbedderTrainer
  • M3
    • Arguments
      • EncoderOnlyEmbedderM3ModelArguments
      • EncoderOnlyEmbedderM3TrainingArguments
    • Modeling
      • EncoderOnlyEmbedderM3Model
        • EncoderOnlyEmbedderM3Model
        • Methods
          • EncoderOnlyEmbedderM3Model.encode()
          • EncoderOnlyEmbedderM3Model.compute_score()
          • EncoderOnlyEmbedderM3Model.compute_dense_score()
          • EncoderOnlyEmbedderM3Model.compute_sparse_score()
          • EncoderOnlyEmbedderM3Model.compute_colbert_score()
          • EncoderOnlyEmbedderM3Model.ensemble_score()
          • EncoderOnlyEmbedderM3Model.forward()
          • EncoderOnlyEmbedderM3Model.compute_loss()
          • EncoderOnlyEmbedderM3Model.gradient_checkpointing_enable()
          • EncoderOnlyEmbedderM3Model.enable_input_require_grads()
          • EncoderOnlyEmbedderM3Model.save()
          • EncoderOnlyEmbedderM3Model._dense_embedding()
          • EncoderOnlyEmbedderM3Model._sparse_embedding()
          • EncoderOnlyEmbedderM3Model._colbert_embedding()
          • EncoderOnlyEmbedderM3Model._encode()
          • EncoderOnlyEmbedderM3Model._compute_similarity()
          • EncoderOnlyEmbedderM3Model._get_queries_attention_mask()
      • EncoderOnlyEmbedderM3ModelForInference
        • EncoderOnlyEmbedderM3ModelForInference
          • EncoderOnlyEmbedderM3ModelForInference.forward()
    • Runner
      • EncoderOnlyEmbedderM3Runner
        • EncoderOnlyEmbedderM3Runner.get_model()
        • EncoderOnlyEmbedderM3Runner.load_tokenizer_and_model()
        • EncoderOnlyEmbedderM3Runner.load_trainer()
    • Trainer
      • EncoderOnlyEmbedderM3Trainer
• Decoder Only
  • Base
    • Arguments
      • DecoderOnlyEmbedderModelArguments
    • Modeling
      • CrossDecoderModel
      • Methods
        • CrossDecoderModel.encode()
    • Runner
      • DecoderOnlyRerankerRunner
        • DecoderOnlyRerankerRunner.load_tokenizer_and_model()
        • DecoderOnlyRerankerRunner.load_trainer()
        • DecoderOnlyRerankerRunner.run()
    • Trainer
      • DecoderOnlyRerankerTrainer
  • ICL
    • Arguments
      • DecoderOnlyEmbedderICLModelArguments
      • DecoderOnlyEmbedderICLDataArguments
    • Dataset
      • DecoderOnlyEmbedderICLSameDatasetTrainDataset
        • DecoderOnlyEmbedderICLSameDatasetTrainDataset
        • Methods
          • DecoderOnlyEmbedderICLSameDatasetTrainDataset._create_batch_data()
      • AbsEmbedderSameDatasetCollator
        • AbsEmbedderSameDatasetCollator
    • Modeling
      • BiDecoderOnlyEmbedderICLModel
      • Methods
        • BiDecoderOnlyEmbedderICLModel.encode()
        • BiDecoderOnlyEmbedderICLModel.compute_score()
        • BiDecoderOnlyEmbedderICLModel.compute_loss()
        • BiDecoderOnlyEmbedderICLModel.gradient_checkpointing_enable()
        • BiDecoderOnlyEmbedderICLModel.enable_input_require_grads()
        • BiDecoderOnlyEmbedderICLModel.save()
        • BiDecoderOnlyEmbedderICLModel._sentence_embedding()
        • BiDecoderOnlyEmbedderICLModel._compute_similarity()
    • Runner
      • DecoderOnlyEmbedderICLRunner
        • DecoderOnlyEmbedderICLRunner.load_tokenizer_and_model()
        • DecoderOnlyEmbedderICLRunner.load_train_dataset()
        • DecoderOnlyEmbedderICLRunner.load_trainer()
        • DecoderOnlyEmbedderICLRunner.run()
    • Trainer
      • DecoderOnlyEmbedderICLTrainer