Views
No views yet
Notice: Licensed by NVIDIA Corporation under the NVIDIA Nemotron Open Model License.
1from transformers import AutoModelForCausalLM, AutoTokenizer
2from peft import PeftModel
3
4# Load base model and tokenizer
5base_model_id = "nvidia/NVIDIA-Nemotron-3-Nano-30B-A3B-Base-BF16"
6adapter_id = "YOUR_USERNAME/nemotron-nano-medmcqa-lora" # Update with your repo
7
8tokenizer = AutoTokenizer.from_pretrained(base_model_id)
9base_model = AutoModelForCausalLM.from_pretrained(
10 base_model_id,
11 torch_dtype="auto",
12 device_map="auto",
13)
14
15# Load LoRA adapter
16model = PeftModel.from_pretrained(base_model, adapter_id)
17
18# Example usage
19prompt = """<|im_start|>system
20You are a medical expert. Answer the multiple choice question.<|im_end|}
21<|im_start|>user
22Question: Which drug causes cinchonism?
23
24A) Aspirin
25B) Quinine
26C) Paracetamol
27D) Ibuprofen<|im_end|>
28<|im_start|>assistant
29"""
30
31inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
32outputs = model.generate(**inputs, max_new_tokens=256)
33print(tokenizer.decode(outputs[0], skip_special_tokens=True))| Parameter | Value |
|---|---|
| LoRA Rank (r) | 16 |
| LoRA Alpha | 32 |
| LoRA Dropout | 0.05 |
| Target Modules | q_proj, k_proj, v_proj, o_proj, up_proj, down_proj |
| Max Sequence Length | 1024 |
| Training Precision | BF16 |
| Optimizer | Unsloth optimized |
1{
2 "r": 16,
3 "lora_alpha": 32,
4 "lora_dropout": 0.05,
5 "target_modules": ["q_proj", "k_proj", "v_proj", "o_proj", "up_proj", "down_proj"],
6 "task_type": "CAUSAL_LM",
7 "bias": "none"
8}1@misc{nemotron3nano,
2 title={NVIDIA Nemotron-3-Nano-30B},
3 author={NVIDIA},
4 year={2025},
5 url={https://huggingface.co/nvidia/NVIDIA-Nemotron-3-Nano-30B-A3B-Base-BF16}
6}
7
8@article{pal2022medmcqa,
9 title={MedMCQA: A Large-scale Multi-Subject Multi-Choice Dataset for Medical domain Question Answering},
10 author={Pal, Ankit and Umapathi, Logesh Kumar and Sankarasubbu, Malaikannan},
11 journal={arXiv preprint arXiv:2203.14371},
12 year={2022}
13}