PEFT 文档

AutoPeftModels

Hugging Face's logo
加入 Hugging Face 社区

并获得增强的文档体验

开始使用

AutoPeftModels

AutoPeftModel 类通过从配置文件中自动推断任务类型来加载相应的 PEFT 模型。它们旨在通过单行代码快速轻松地加载 PEFT 模型,而无需担心需要哪个确切的模型类或手动加载 PeftConfig

AutoPeftModel

class peft.AutoPeftModel

< >

( *args **kwargs )

from_pretrained

< >

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

这是一个包装器,包含了用户加载 PEFT 模型所需执行的所有预处理步骤。kwargs 会传递给 PeftConfig,它会自动处理 Hub 方法的 kwargs 和配置对象初始化的过滤。

AutoPeftModelForCausalLM

class peft.AutoPeftModelForCausalLM

< >

( *args **kwargs )

AutoPeftModelForSeq2SeqLM

class peft.AutoPeftModelForSeq2SeqLM

< >

( *args **kwargs )

AutoPeftModelForSequenceClassification

class peft.AutoPeftModelForSequenceClassification

< >

( *args **kwargs )

AutoPeftModelForTokenClassification

class peft.AutoPeftModelForTokenClassification

< >

( *args **kwargs )

AutoPeftModelForQuestionAnswering

class peft.AutoPeftModelForQuestionAnswering

< >

( *args **kwargs )

AutoPeftModelForFeatureExtraction

class peft.AutoPeftModelForFeatureExtraction

< >

( *args **kwargs )

< > 在 GitHub 上更新