Fine-tuned LoRA adapter on
meta-llama/Llama-3.3-70B-Instruct
for Indian medical entrance examination QA (AIIMS-PG and NEET-PG).
Benchmark:
openlifescienceai/medmcqa
validation set — 4,183 questions across 21 medical subjects.
38,000+ instruction-formatted NEET-PG and AIIMS-PG questions processed
via Adaption Labs Adaptive Data with reasoning traces, hallucination
mitigation, and Indian medical context localization.
1from peft import PeftModel
2from transformers import AutoModelForCausalLM, AutoTokenizer
3import torch
4
5base = AutoModelForCausalLM.from_pretrained(
6 "meta-llama/Llama-3.3-70B-Instruct",
7 torch_dtype=torch.float16,
8 device_map="auto"
9)
10tokenizer = AutoTokenizer.from_pretrained("xenkrypt/MedLlama-India-70B")
11model = PeftModel.from_pretrained(base, "xenkrypt/MedLlama-India-70B")
12
13prompt = """### Instruction:
14You are a medical expert for AIIMS/NEET-PG examinations.
15Answer this multiple choice question.
16
17Question: Most common cause of mitral stenosis?
18A) Rheumatic fever
19B) Infective endocarditis
20C) Congenital
21D) SLE
22
23### Response:
24The correct answer is"""
25
26inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
27out = model.generate(**inputs, max_new_tokens=150, do_sample=False)
28print(tokenizer.decode(out[0], skip_special_tokens=True))