DrugGen — Hardware Requirements & GPU Compatibility
Chat
156 downloads
4 likesSpecifications
- Publisher
- alimotahharynia
- Parameters
- 127M
- Architecture
- GPT2LMHeadModel
- Context Length
- 1,024 tokens
- Vocabulary Size
- 53,083
- Release Date
- 2025-05-22
- License
- GPL 3.0
Get Started
HuggingFace
How Much VRAM Does DrugGen Need?
Select a quantization to see compatible GPUs below.
| Quantization | Bits | VRAM | + Context | File Size | Quality |
|---|---|---|---|---|---|
| BF16 | 16.00 | 0.3 GB | — | 0.25 GB | Brain floating point 16 — preferred for training |
Which GPUs Can Run DrugGen?
BF16 · 0.3 GBDrugGen (BF16) requires 0.3 GB of VRAM to load the model weights. For comfortable inference with headroom for KV cache and system overhead, 1+ GB is recommended. 35 GPUs can run it, including NVIDIA GeForce RTX 5090, NVIDIA GeForce RTX 3090 Ti.
Runs great
— Plenty of headroomWhich Devices Can Run DrugGen?
BF16 · 0.3 GB33 devices with unified memory can run DrugGen, including NVIDIA DGX H100, NVIDIA DGX A100 640GB.
Runs great
— Plenty of headroomRelated Models
Frequently Asked Questions
- How much VRAM does DrugGen need?
DrugGen requires 0.3 GB of VRAM at BF16.
VRAM = Weights + KV Cache + Overhead
Weights = 127M × 16 bits ÷ 8 = 0.3 GB
VRAM usage by quantization
BF160.3 GB- Can I run DrugGen on a Mac?
DrugGen requires at least 0.3 GB at BF16, which exceeds the unified memory of most consumer Macs. You would need a Mac Studio or Mac Pro with a high-memory configuration.
- Can I run DrugGen locally?
Yes — DrugGen can run locally on consumer hardware. At BF16 quantization it needs 0.3 GB of VRAM. Popular tools include Ollama, LM Studio, and llama.cpp.
- How fast is DrugGen?
At BF16, DrugGen can reach ~10411 tok/s on AMD Instinct MI300X. On NVIDIA GeForce RTX 4090: ~2340 tok/s. Speed depends mainly on GPU memory bandwidth. Real-world results typically within ±20%.
tok/s = (bandwidth GB/s ÷ model GB) × efficiency
Example: AMD Instinct MI300X → 5300 ÷ 0.3 × 0.55 = ~10411 tok/s
Estimated speed at BF16 (0.3 GB)
AMD Instinct MI300X~10411 tok/sNVIDIA GeForce RTX 4090~2340 tok/sNVIDIA H100 SXM~7781 tok/sAMD Instinct MI250X~6437 tok/sReal-world results typically within ±20%. Speed depends on batch size, quantization kernel, and software stack.
- What's the download size of DrugGen?
At BF16, the download is about 0.25 GB.
- Which GPUs can run DrugGen?
35 consumer GPUs can run DrugGen at BF16 (0.3 GB). Top options include AMD Radeon RX 6700 XT, AMD Radeon RX 6800, AMD Radeon RX 6800 XT. 35 GPUs have plenty of headroom for comfortable inference.
- Which devices can run DrugGen?
33 devices with unified memory can run DrugGen at BF16 (0.3 GB), including Mac Mini M4 (16 GB), Mac Mini M4 (32 GB), Mac Mini M4 Pro (24 GB), Mac Mini M4 Pro (48 GB). Apple Silicon Macs use unified memory shared between CPU and GPU, making them well-suited for local LLM inference.