TRL documentation
استخدام النموذج بعد التدريب
استخدام النموذج بعد التدريب
بمجرد تدريب نموذج باستخدام SFTTrainer أو DPOTrainer، ستحصل على نموذج مُعدّل يمكن استخدامه لتوليد النص. في هذا القسم، سنستعرض عملية تحميل النموذج المُعدّل وتوليد النص. إذا كنت بحاجة إلى تشغيل خادم استدلال بالنموذج المُدرَّب، يمكنك استكشاف مكتبات مثل text-generation-inference.
التحميل والتوليد
إذا كنت قد قمت بضبط نموذج بالكامل، أي دون استخدام PEFT، يمكنك ببساطة تحميله مثل أي نموذج لغة آخر في transformers:
import torch
from transformers import AutoTokenizer, AutoModelForCausalLM
model_name_or_path = "Qwen/Qwen3-0.6B" #path/to/your/model/or/name/on/hub
device = torch.accelerator.current_accelerator() if torch.accelerator.is_available() else "cpu"
model = AutoModelForCausalLM.from_pretrained(model_name_or_path).to(device)
tokenizer = AutoTokenizer.from_pretrained(model_name_or_path)
inputs = tokenizer.encode("This movie was really", return_tensors="pt").to(device)
outputs = model.generate(inputs)
print(tokenizer.decode(outputs[0]))بديلًا عن ذلك، يمكنك أيضًا استخدام خط الأنابيب:
from transformers import pipeline
model_name_or_path = "Qwen/Qwen3-0.6B" #path/to/your/model/or/name/on/hub
pipe = pipeline("text-generation", model=model_name_or_path)
print(pipe("This movie was really")[0]["generated_text"])استخدام Adapters PEFT
from peft import PeftConfig, PeftModel
from transformers import AutoModelForCausalLM, AutoTokenizer
base_model_name = "Qwen/Qwen3-0.6B" #path/to/your/model/or/name/on/hub
adapter_model_name = "path/to/my/adapter"
model = AutoModelForCausalLM.from_pretrained(base_model_name)
model = PeftModel.from_pretrained(model, adapter_model_name)
tokenizer = AutoTokenizer.from_pretrained(base_model_name)يمكنك أيضًا دمج المُكيّفات (adapters) في النموذج الأساسي بحيث تتمكن من استخدام النموذج كنموذج transformers عادي، غير أن نقطة الحفظ (checkpoint) ستكون أكبر بكثير:
model = AutoModelForCausalLM.from_pretrained(base_model_name)
model = PeftModel.from_pretrained(model, adapter_model_name)
model = model.merge_and_unload()
model.save_pretrained("merged_adapters")بمجرد تحميل النموذج، سواء بعد دمج المُكيّفات أو إبقائها منفصلة فوقه، يمكنك تشغيل التوليد كما هو موضح أعلاه بالنسبة لنموذج عادي.
Update on GitHub