AutoPeftModels

Kelas AutoPeftModel memuat model PEFT yang sesuai untuk jenis tugas dengan cara menyimpulkan secara otomatis dari file konfigurasi. Kelas-kelas ini dirancang untuk memuat model PEFT dengan cepat dan mudah dalam satu baris kode, tanpa perlu khawatir tentang kelas model spesifik yang dibutuhkan atau memuat PeftConfig secara manual.

AutoPeftModel

kelas peft.AutoPeftModel

< source >

( *args**kwargs )

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 )

Parameter

import_allowlist (list[str], opsional, defaultnya {get_default_import_allowlist()}) — AutoPeftModel akan mencoba menginstansiasi model dasar yang dikonfigurasi dalam konfigurasi adapter. Karena operasi ini berpotensi mengimpor paket lain, allowlist ini berfungsi sebagai pengaman untuk mencegah impor paket berbahaya. Anda mungkin perlu menentukan nama impor paket Anda di sini jika tidak termasuk dalam nilai default.

Wrapper untuk semua langkah pra-pemrosesan yang perlu dilakukan pengguna untuk memuat model PEFT. Argumen kwargs diteruskan ke PeftConfig yang secara otomatis menangani penyaringan kwargs dari metode Hub dan inisialisasi objek config.

Parameter-parameter ini setara dengan parameter dari PeftModel.from_pretrained(). Perbedaannya didokumentasikan di bawah.