Fine-tuned version of
Qwen2.5-1.5B-Instruct
on the
MedMCQA dataset using QLoRA.
MedMCQA contains ~182k multiple-choice questions from AIIMS and NEET-PG Indian medical entrance
exams across 23 subjects. Only single-choice questions were used for training and evaluation.
Note: Dental accounts for 30% of the evaluation set and is the weakest subject,
reflecting limited dental knowledge in the base model's pre-training.
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model = AutoModelForCausalLM.from_pretrained("Satya-Dey/qwen2.5-1.5b-medmcqa-qlora-v2")
4tokenizer = AutoTokenizer.from_pretrained("Satya-Dey/qwen2.5-1.5b-medmcqa-qlora-v2")
5
6prompt = (
7 "### Question:\n"
8 "Commonest site of aortic aneurysm?\n"
9 "A) Ascending aorta B) Arch of aorta "
10 "C) Descending thoracic aorta D) Infrarenal abdominal aorta\n\n"
11 "### Answer:\n"
12)
13
14inputs = tokenizer(prompt, return_tensors="pt")
15outputs = model.generate(**inputs, max_new_tokens=5, do_sample=False)
16print(tokenizer.decode(outputs[0], skip_special_tokens=True))