Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2from peft import PeftModel
3
4model = AutoModelForCausalLM.from_pretrained("MNLP_M3_mcqa_model_optimized")
5tokenizer = AutoTokenizer.from_pretrained("MNLP_M3_mcqa_model_optimized")
6
7# Example inference
8question = "What is the capital of France?"
9choices = ["London", "Berlin", "Paris", "Madrid"]
10
11prompt = f"Question: {question}\nA) {choices[0]}\nB) {choices[1]}\nC) {choices[2]}\nD) {choices[3]}\nAnswer:"
12inputs = tokenizer(prompt, return_tensors="pt")
13outputs = model.generate(**inputs, max_new_tokens=5, temperature=0.1)
14answer = tokenizer.decode(outputs[0][inputs['input_ids'].shape[1]:], skip_special_tokens=True).strip()