Upload 7 files 8cf5deb verified
Oscar Wang commited on
How to use saneowl/test-001 with PEFT:
from peft import PeftModel
from transformers import AutoModelForCausalLM
base_model = AutoModelForCausalLM.from_pretrained("google/gemma-2b")
model = PeftModel.from_pretrained(base_model, "saneowl/test-001")