Prepare Model • AI Agents Framework

prepare_model: API reference for TrainModel.prepare_model

prepare_model

Method

</div>

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

Signature

python
def prepare_model() -> Any

Returns

ResponseField 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