How to use sumith2425/ppo-prefix-model 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, "sumith2425/ppo-prefix-model")
6560429 1cfda0b 6560429
1
2
3
4
version https://git-lfs.github.com/spec/v1 oid sha256:84851e10beaf2094f5da4a5d30776798578403e684577b8f99fde367e52634eb size 6037
