Views
No views yet
AnnaelleMyriam/SFT_M3_modelaymanbakiri/MNLP_M3_mcqa_model_adapterAnnaelleMyriam/SFT_M3_model1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model = AutoModelForCausalLM.from_pretrained("aymanbakiri/MNLP_M3_mcqa_model1")
4tokenizer = AutoTokenizer.from_pretrained("aymanbakiri/MNLP_M3_mcqa_model1")
5
6# Example usage for MCQA
7prompt = """Question: What is the capital of France?
8Options: (A) London (B) Berlin (C) Paris (D) Madrid
9Answer:"""
10
11inputs = tokenizer(prompt, return_tensors="pt")
12outputs = model.generate(**inputs, max_new_tokens=5)
13answer = tokenizer.decode(outputs[0], skip_special_tokens=True)
14print(answer)