modalities.conversion package

Subpackages

• modalities.conversion.gpt2 package
  • Submodules
  • modalities.conversion.gpt2.configuration_gpt2 module
    • GPT2Config
      • GPT2Config.base_model_tp_plan
      • GPT2Config.keys_to_ignore_at_inference
      • GPT2Config.model_type
  • modalities.conversion.gpt2.conversion_code module
    • transfer_model_code()
  • modalities.conversion.gpt2.conversion_model module
    • check_converted_model()
    • convert_model_checkpoint()
    • convert_model_config()
  • modalities.conversion.gpt2.conversion_tokenizer module
    • convert_tokenizer()
  • modalities.conversion.gpt2.convert_gpt2 module
    • convert_gpt2()
  • modalities.conversion.gpt2.modeling_gpt2 module
    • GPT2DecoderLayer
      • GPT2DecoderLayer.forward()
    • GPT2ForCausalLM
      • GPT2ForCausalLM.forward()
      • GPT2ForCausalLM.get_decoder()
      • GPT2ForCausalLM.get_input_embeddings()
      • GPT2ForCausalLM.get_output_embeddings()
      • GPT2ForCausalLM.set_decoder()
      • GPT2ForCausalLM.set_input_embeddings()
      • GPT2ForCausalLM.set_output_embeddings()
    • GPT2ForQuestionAnswering
      • GPT2ForQuestionAnswering.base_model_prefix
      • GPT2ForQuestionAnswering.forward()
      • GPT2ForQuestionAnswering.get_input_embeddings()
      • GPT2ForQuestionAnswering.set_input_embeddings()
    • GPT2ForSequenceClassification
      • GPT2ForSequenceClassification.forward()
      • GPT2ForSequenceClassification.get_input_embeddings()
      • GPT2ForSequenceClassification.set_input_embeddings()
    • GPT2ForTokenClassification
      • GPT2ForTokenClassification.forward()
      • GPT2ForTokenClassification.get_input_embeddings()
      • GPT2ForTokenClassification.set_input_embeddings()
    • GPT2Model
      • GPT2Model.forward()
      • GPT2Model.get_input_embeddings()
      • GPT2Model.set_input_embeddings()
    • GPT2PreTrainedModel
      • GPT2PreTrainedModel.base_model_prefix
      • GPT2PreTrainedModel.config_class
      • GPT2PreTrainedModel.supports_gradient_checkpointing
    • KwargsForCausalLM
      • KwargsForCausalLM.cumulative_seqlens_k
      • KwargsForCausalLM.cumulative_seqlens_q
      • KwargsForCausalLM.max_length_k
      • KwargsForCausalLM.max_length_q
      • KwargsForCausalLM.num_items_in_batch
    • LlamaAttention
      • LlamaAttention.forward()
    • LlamaDynamicNTKScalingRotaryEmbedding
    • LlamaFlashAttention2
      • LlamaFlashAttention2.forward()
    • LlamaLinearScalingRotaryEmbedding
    • LlamaMLP
      • LlamaMLP.forward()
    • LlamaRotaryEmbedding
      • LlamaRotaryEmbedding.forward()
    • LlamaSdpaAttention
      • LlamaSdpaAttention.forward()
    • apply_rotary_pos_emb()
    • repeat_kv()
    • rotate_half()
  • Module contents

Module contents