mainlp
/

MCQ-Classifier-MMLU-XYZ

Model card Files Files and versions Community

xinpeng commited on May 16

Commit

f345785

•

1 Parent(s): 41bba5e

update

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -38,7 +38,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 from peft import PeftModel, PeftConfig
 config = PeftConfig.from_pretrained("mainlp/MCQ-Classifier-MMLU-XYZ")
 base_model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1")
-model = PeftModel.from_pretrained(base_model, "mainlp/MCQ-Classifier")
 tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-v0.1")
 to_classify = f"""<s>[INST] Classify the response.{inputs} [/INST]"""
 model_input = tokenizer(to_classify, return_tensors="pt")

 from peft import PeftModel, PeftConfig
 config = PeftConfig.from_pretrained("mainlp/MCQ-Classifier-MMLU-XYZ")
 base_model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1")
+model = PeftModel.from_pretrained(base_model, "mainlp/MCQ-Classifier-MMLU-XYZ")
 tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-v0.1")
 to_classify = f"""<s>[INST] Classify the response.{inputs} [/INST]"""
 model_input = tokenizer(to_classify, return_tensors="pt")