标题类的继承关系如下:
PreTrainedModel -> LlamaPreTrainedModel -> LlamaModel -> LlamaForCausalLM.
1 PretrainedModel
PretrainedModel继承自:
2 LlamaPreTrainedModel
必须初始化的属性和方法如下:
2.1 Llama
2.2 Qwen2
3 LlamaModel
必须重写的方法:
- __init__()
- get_input_embedding()
- set_input_embedding()
- forward()
3.1 QwenModel
4 LlamaForCasualLM
必须重写的方法:
- __init__()
- get_input_embedding()
- set_input_embedding()
- get_output_embedding()
- set_output_embedding()
- get_decoder()
- set_decoder()
- forward()
- prepare_input_for_generation()
- _reorder_cache()
4.1 Qwen