Views
No views yet

from peft import PeftModel, PeftConfig
from transformers import AutoModelForCausalLM, AutoTokenizer
config = PeftConfig.from_pretrained("therealcyberlord/magicwand-gptneo-1.3b")
model = AutoModelForCausalLM.from_pretrained("EleutherAI/gpt-neo-1.3B")
model = PeftModel.from_pretrained(model, "therealcyberlord/magicwand-gptneo-1.3b")
tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-neo-1.3B")