Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2from peft import PeftModel
3
4model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1", device_map="auto")
5tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-v0.1")
6model = PeftModel.from_pretrained(model, "frankmorales2020/mistral-7b-qwen-Next-80B-A3B-Instruct-distilled")