therpist2 / app.py
hackergeek98's picture
Create app.py
b6c5a4f verified
raw
history blame
225 Bytes
from peft import PeftModel
from transformers import AutoModelForCausalLM
base_model = AutoModelForCausalLM.from_pretrained("google/gemma-3-1b-pt")
model = PeftModel.from_pretrained(base_model, "hackergeek98/gemma-finetuned")