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
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.