hieunguyenminh/roleplay
Viewer • Updated • 5.76k • 305 • 92
How to use kingabzpro/gemma-7b-it-v2-role-play with PEFT:
from peft import PeftModel
from transformers import AutoModelForCausalLM
base_model = AutoModelForCausalLM.from_pretrained("google/gemma-7b-it")
model = PeftModel.from_pretrained(base_model, "kingabzpro/gemma-7b-it-v2-role-play")This model is a fine-tuned version of google/gemma-7b-it on an hieunguyenminh/roleplay dataset.