Upload 16 files 6eb46c5 verified
Sri Santh commited on
How to use SriSanthM/Mistral-Docker 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, "SriSanthM/Mistral-Docker")