SOTOPIA
Collection
9 items • Updated • 6
How to use cmu-lti/sotopia-pi-mistral-7b-BC_SR with PEFT:
from peft import PeftModel
from transformers import AutoModelForCausalLM
base_model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-Instruct-v0.1")
model = PeftModel.from_pretrained(base_model, "cmu-lti/sotopia-pi-mistral-7b-BC_SR")https://arxiv.org/pdf/2403.08715
The following bitsandbytes quantization config was used during training: