Hugging Face's logo

mainlp
/
MCQ-Classifier-MMLU-XYZ

PEFT
Safetensors
Model card Files
xet
 Community
xinpeng commited on
Commit
f345785
·
verified ·
1 Parent(s): 41bba5e

update

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -38,7 +38,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
38
  from peft import PeftModel, PeftConfig
39
  config = PeftConfig.from_pretrained("mainlp/MCQ-Classifier-MMLU-XYZ")
40
  base_model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1")
41
- model = PeftModel.from_pretrained(base_model, "mainlp/MCQ-Classifier")
42
  tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-v0.1")
43
  to_classify = f"""<s>[INST] Classify the response.{inputs} [/INST]"""
44
  model_input = tokenizer(to_classify, return_tensors="pt")
 
38
  from peft import PeftModel, PeftConfig
39
  config = PeftConfig.from_pretrained("mainlp/MCQ-Classifier-MMLU-XYZ")
40
  base_model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1")
41
+ model = PeftModel.from_pretrained(base_model, "mainlp/MCQ-Classifier-MMLU-XYZ")
42
  tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-v0.1")
43
  to_classify = f"""<s>[INST] Classify the response.{inputs} [/INST]"""
44
  model_input = tokenizer(to_classify, return_tensors="pt")