AutoPeftModels

The AutoPeftModel classes loads the appropriate PEFT model for the task type by automatically inferring it from the configuration file. They are designed to quickly and easily load a PEFT model in a single line of code without having to worry about which exact model class you need or manually loading a PeftConfig.

AutoPeftModel

class peft.AutoPeftModel

< source >

( *args **kwargs )

from_pretrained

< source >

( pretrained_model_name_or_path adapter_name: str = 'default' is_trainable: bool = False config: Optional[PeftConfig] = None revision: Optional[str] = None **kwargs )

A wrapper around all the preprocessing steps a user needs to perform in order to load a PEFT model. The kwargs are passed along to PeftConfig that automatically takes care of filtering the kwargs of the Hub methods and the config object init.