Downloads · 30 days
0
QuantumindSSI/09-drug-interaction-predictor
09-drug-interaction-predictor is a text generation model from QuantumindSSI. Use it when you need the model to write or continue text. It is set up for transformers. The card lists the license as apache-2.0.
A fine-tuned Small Language Model (SLM) that predicts potential drug-drug interactions from medication lists, including severity, mechanism, evidence, and clinical recommendations.
Downloads · 30 days
0
Access
Public
Updated Aug 11, 2026
Repo size
46.7 MB
Likes
0
Public
Click a slice to open those files.
.safetensors35.2 MB · 76%
From the Hugging Face model README
A fine-tuned Small Language Model (SLM) that predicts potential drug-drug interactions from medication lists, including severity, mechanism, evidence, and clinical recommendations.
| Attribute | Value |
|---|---|
| Developer | QuantumIndSSI Ltd |
| Base Model | Qwen/Qwen2-0.5B-Instruct |
| Architecture | Transformer decoder (causal LM) |
| Fine-tuning Method | LoRA (Low-Rank Adaptation) |
| LoRA Rank | 16 |
| LoRA Alpha | 32 |
| License | apache-2.0 |
model_id = "QuantumindSSI/09_drug_interaction_predictor"
from transformers import AutoModelForCausalLM, AutoTokenizer
model = AutoModelForCausalLM.from_pretrained(model_id)
tokenizer = AutoTokenizer.from_pretrained(model_id)
prompt = """Medications: Warfarin, Aspirin, Omeprazole. Analyze for interactions."""
inputs = tokenizer(prompt, return_tensors="pt")
outputs = model.generate(**inputs, max_new_tokens=512)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))
| Target | RAM | Notes |
|---|---|---|
| Cloud GPU | 4GB | FP16 inference |
| Workstation | 3GB | INT8 quantized |
| Victron Edge | 2-3GB | INT8/INT4 quantized, CPU |
@misc{09_drug_interaction_predictor,
title={Drug Interaction Predictor},
author={QuantumIndSSI Ltd},
year={2026},
publisher={Hugging Face},
howpublished={\url{https://huggingface.co/QuantumindSSI/09_drug_interaction_predictor}}
}