mainlp
/

MCQ-Classifier-MMLU-EFG

Model card Files Files and versions Community

xinpeng commited on May 16

Commit

e5dd436

•

1 Parent(s): 8a5cfe0

update readme

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -37,7 +37,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 from peft import PeftModel, PeftConfig
 config = PeftConfig.from_pretrained("mainlp/MCQ-Classifier-MMLU-EFG")
 base_model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1")
-model = PeftModel.from_pretrained(base_model, "mainlp/MCQ-Classifier-EFG")
 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-EFG")
 base_model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1")
+model = PeftModel.from_pretrained(base_model, "mainlp/MCQ-Classifier--MMLU-EFG")
 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")