PEFT documentation
AutoPeftModels
AutoPeftModels
Les classes AutoPeftModel chargent le modèle PEFT approprié pour le type de tâche en le déduisant automatiquement du fichier de configuration. Elles sont conçues pour charger rapidement et facilement un modèle PEFT en une seule ligne de code, sans avoir à se soucier de la classe de modèle exacte requise ni de charger manuellement un PeftConfig.
AutoPeftModel
from_pretrained
< source >( pretrained_model_name_or_pathadapter_name: str = 'default'is_trainable: bool = Falseconfig: Optional[PeftConfig] = Nonerevision: Optional[str] = Noneimport_allowlist: Optional[list[str]] = None**kwargs )
Paramètres
- import_allowlist (
list[str], facultatif, par défaut{get_default_import_allowlist()}) — AutoPeftModel tentera d’instancier le modèle de base configuré dans la configuration de l’adaptateur. Comme cette opération peut nécessiter l’importation d’autres paquets, cette liste blanche sert de garde-fou pour empêcher l’importation de paquets malveillants. Vous devrez peut-être spécifier ici le nom d’importation de votre paquet s’il ne figure pas dans les valeurs par défaut.
Un wrapper autour de toutes les étapes de prétraitement qu'un utilisateur doit effectuer pour charger un modèle PEFT. Les kwargs sont transmis à PeftConfig, qui prend automatiquement en charge le filtrage des kwargs des méthodes du Hub et de l'initialisation de l'objet de configuration.
Les paramètres sont équivalents à ceux de PeftModel.from_pretrained(). Les différences sont documentées ci-dessous.