Hub documentation

استخدام PEFT في Hugging Face

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

استخدام PEFT في Hugging Face

🤗 Parameter-Efficient Fine-Tuning (PEFT) هي مكتبة لتكييف نماذج اللغة المُدرَّبة مسبقًا بكفاءة على تطبيقات مختلفة دون الحاجة إلى ضبط جميع معاملات النموذج.

استكشاف PEFT على Hub

يمكنك العثور على نماذج PEFT من خلال التصفية على يسار صفحة النماذج.

التثبيت

للبدء، يمكنك الاطلاع على الجولة السريعة في وثائق PEFT. وللتثبيت، اتبع دليل تثبيت PEFT. يمكنك أيضًا استخدام أمر التثبيت التالي من سطر واحد عبر pip:

$ pip install peft

استخدام نماذج موجودة

يمكن تحميل جميع نماذج PEFT من Hub. وللاستفادة من نموذج PEFT، يجب أيضًا تحميل النموذج الأساسي الذي تم ضبطه، كما هو موضح أدناه. يحتوي كل نموذج مُضبط على النموذج الأساسي في بطاقة النموذج الخاصة به.

from transformers import AutoModelForCausalLM, AutoTokenizer
from peft import PeftModel, PeftConfig

base_model = "mistralai/Mistral-7B-v0.1"
adapter_model = "dfurman/Mistral-7B-Instruct-v0.2"

model = AutoModelForCausalLM.from_pretrained(base_model)
model = PeftModel.from_pretrained(model, adapter_model)
tokenizer = AutoTokenizer.from_pretrained(base_model)

model = model.to("cuda")
model.eval()

بمجرد التحميل، يمكنك تمرير مدخلاتك إلى مُرمّز الرموز (tokenizer) لتحضيرها، ثم استدعاء model.generate() بالطريقة المعتادة في transformers.

inputs = tokenizer("Tell me the recipe for chocolate chip cookie", return_tensors="pt")

with torch.no_grad():
    outputs = model.generate(input_ids=inputs["input_ids"].to("cuda"), max_new_tokens=10)
    print(tokenizer.batch_decode(outputs.detach().cpu().numpy(), skip_special_tokens=True)[0])

يُخرج ما يلي:

Tell me the recipe for chocolate chip cookie dough.

1. Preheat oven to 375 degrees F (190 degrees C).
2. In a large bowl, cream together 1/2 cup (1 stick) of butter or margarine, 1/2 cup granulated sugar, and 1/2 cup packed brown sugar.
3. Beat in 1 egg and 1 teaspoon vanilla extract.
4. Mix in 1 1/4 cups all-purpose flour.
5. Stir in 1/2 teaspoon baking soda and 1/2 teaspoon salt.
6. Fold in 3/4 cup semisweet chocolate chips.
7. Drop by

إذا كنت ترغب في تحميل نموذج PEFT محدد، يمكنك النقر على Use in PEFT في بطاقة النموذج وستحصل على مقتطف برمجي جاهز للاستخدام!

موارد إضافية

Update on GitHub