AG-Prompt- Unified Multilingual-Trained-Models
Collection
4 items • Updated
How to use surrey-nlp/AG-openchat with PEFT:
from peft import PeftModel
from transformers import AutoModelForCausalLM
base_model = AutoModelForCausalLM.from_pretrained("openchat/openchat-3.5-0106")
model = PeftModel.from_pretrained(base_model, "surrey-nlp/AG-openchat")This model is a fine-tuned version of openchat/openchat-3.5-0106 on using AG prompt.
The following hyperparameters were used during training: