PEFT documentation

AutoPeftModels

You are viewing main version, which requires installation from source. If you'd like regular pip install, checkout the latest stable version (v0.21.0).
Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

AutoPeftModels

AutoPeftModel クラスは、設定ファイルからタスクの種類を自動的に推論し、適切な PEFT モデルを読み込みます。これらは、どのモデルクラスが必要かや、PeftConfig を手動で読み込む必要を気にすることなく、1 行のコードで PEFT モデルを素早く簡単に読み込むことを目的に設計されています。

AutoPeftModel

クラス peft.AutoPeftModel

< >

( *args**kwargs )

from_pretrained

< >

( pretrained_model_name_or_pathadapter_name: str = 'default'is_trainable: bool = Falseconfig: Optional[PeftConfig] = Nonerevision: Optional[str] = Noneimport_allowlist: Optional[list[str]] = None**kwargs )

パラメータ

  • import_allowlist (list[str]、任意、デフォルトは {get_default_import_allowlist()}) — AutoPeftModel は、アダプター設定で設定されているベースモデルのインスタンス化を試みます。 この操作では他のパッケージをインポートする可能性があるため、この許可リストは悪意のあるパッケージのインポートを防ぐための安全策です。 デフォルトにない場合は、ここでパッケージのインポート名を指定する必要がある場合があります。

PEFTモデルを読み込むためにユーザーが実行する必要があるすべての前処理ステップをラップするクラスです。kwargsはPeftConfigに渡され、Hubメソッドのkwargsとconfigオブジェクトのinitのフィルタリングが自動的に処理されます。

パラメータはPeftModel.from_pretrained()のものと同等です。違いについては以下に記載しています。

AutoPeftModelForCausalLM

クラス peft.AutoPeftModelForCausalLM

< >

( *args**kwargs )

AutoPeftModelForSeq2SeqLM

クラス peft.AutoPeftModelForSeq2SeqLM

< >

( *args**kwargs )

AutoPeftModelForSequenceClassification

クラス peft.AutoPeftModelForSequenceClassification

< >

( *args**kwargs )

AutoPeftModelForTokenClassification

クラス peft.AutoPeftModelForTokenClassification

< >

( *args**kwargs )

AutoPeftModelForQuestionAnswering

クラス peft.AutoPeftModelForQuestionAnswering

< >

( *args**kwargs )

AutoPeftModelForFeatureExtraction

クラス peft.AutoPeftModelForFeatureExtraction

< >

( *args**kwargs )

Update on GitHub