Prepare Model • AI Agents Framework

prepare_model: API reference for TrainVisionModel.prepare_model

prepare_model

Method

</div>

This is a method of the TrainVisionModel class in the train_vision module.

Signature

python
def prepare_model() -> Any

Returns

ResponseField Any

The result of the operation.

Uses

  • FastVisionModel.from_pretrained
  • FastVisionModel.get_peft_model

Used By

Source

View on GitHub

praisonai/train_vision.py at line 60