How to use secllmuser/constitutional-toxic-classifier-gemma with PEFT:
from peft import PeftModel from transformers import AutoModelForSequenceClassification base_model = AutoModelForSequenceClassification.from_pretrained("google/gemma-2b") model = PeftModel.from_pretrained(base_model, "secllmuser/constitutional-toxic-classifier-gemma")
