Skip to main content

prepare_model

Method
This is a method of the TrainModel class in the train module.

Signature

def prepare_model() -> Any

Returns

Returns
Any
The result of the operation.

Uses

  • FastLanguageModel.from_pretrained
  • get_chat_template
  • FastLanguageModel.get_peft_model

Used By

Source

View on GitHub

praisonai/train.py at line 143