Downloads · 30 days
9
23% of all-time downloads
demo-ddi-1.5b-lora
amank-root/demo-ddi-1.5b-lora
demo-ddi-1.5b-lora is a machine learning model from amank-root. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for peft.
LoRA adapter fine-tuned on the OpenEnv-DDI drug-drug interaction triage task.
All-time downloads
40
Public
Repo size
125 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors111 MB · 60%
At a glance
- Library
- peft
- Access
- Public
- Created
- Jul 13, 2026
- Updated
- Jul 13, 2026
- SHA
- 6b813000
Base models
Model card
From the Hugging Face model README
DDI Triage LoRA — amank-root/demo-ddi-1.5b-lora
LoRA adapter fine-tuned on the OpenEnv-DDI drug-drug interaction triage task.
Base model: Qwen/Qwen2.5-1.5B-Instruct
Task: Given a patient's medication list and labs, flag/monitor/ignore DDIs and suggest safer alternatives.
Training: 2 500 SFT rows (18 real + 480 synthetic cases), 2 epochs, r=8 LoRA.
Load
from transformers import AutoModelForCausalLM, AutoTokenizer
from peft import PeftModel
base = AutoModelForCausalLM.from_pretrained("Qwen/Qwen2.5-1.5B-Instruct", torch_dtype="auto", device_map="auto")
tokenizer = AutoTokenizer.from_pretrained("amank-root/demo-ddi-1.5b-lora")
model = PeftModel.from_pretrained(base, "amank-root/demo-ddi-1.5b-lora")
model = model.merge_and_unload() # optional: merge for faster inference
- Library
- peft
- Created
- Jul 13, 2026
- Updated
- Jul 13, 2026