How to use Amogh06/PPO-Prefix-SafeGenAI with PEFT:
from peft import PeftModel from transformers import AutoModelForCausalLM base_model = AutoModelForCausalLM.from_pretrained("openai-community/gpt2-medium") model = PeftModel.from_pretrained(base_model, "Amogh06/PPO-Prefix-SafeGenAI")
Amogh06 commited on 