Views
No views yet
mgatti/MNLP_M3_mcqa_model - Multiple-choice QA capabilitiesalbertfares/MNLP_M3_dpo_model - Preference-aligned responses1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model = AutoModelForCausalLM.from_pretrained("merged_mnlp_m3_sft_dpo")
4tokenizer = AutoTokenizer.from_pretrained("merged_mnlp_m3_sft_dpo")
5
6# For MCQA
7prompt = "Which of the following is correct? A) 2+2=5 B) 2+2=4 C) 2+2=3"
8inputs = tokenizer(prompt, return_tensors="pt")
9outputs = model.generate(**inputs, max_length=200)
10print(tokenizer.decode(outputs[0], skip_special_tokens=True))
11
12# For general generation
13prompt = "Explain the concept of recursion in programming"
14inputs = tokenizer(prompt, return_tensors="pt")
15outputs = model.generate(**inputs, max_length=300, temperature=0.7)
16print(tokenizer.decode(outputs[0], skip_special_tokens=True))