PEFT How to use naive-puzzle/reward_output with PEFT:
from peft import PeftModel
from transformers import AutoModelForSequenceClassification
base_model = AutoModelForSequenceClassification.from_pretrained("google/gemma-2-2b-jpn-it")
model = PeftModel.from_pretrained(base_model, "naive-puzzle/reward_output") 
