bmartinc80/gemma-ft 5f3255d verified
Benito Martin commited on
How to use benitomartin/gemma-ft with PEFT:
from peft import PeftModel
from transformers import AutoModelForCausalLM
base_model = AutoModelForCausalLM.from_pretrained("google/gemma-2b")
model = PeftModel.from_pretrained(base_model, "benitomartin/gemma-ft")