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