Skip to content

Installation

From PyPI

pip install model-garage

With Optional Dependencies

# Development tools (pytest, ruff, mypy)
pip install model-garage[dev]

# TUI dashboard (Textual)
pip install model-garage[dashboard]

# Everything
pip install model-garage[all]

From Source

git clone https://github.com/Lumi-node/model-garage
cd model-garage
pip install -e ".[dev]"

Requirements

  • Python 3.10+
  • PyTorch 2.0+
  • Transformers 4.30+

GPU Optional

Model Garage works on CPU for small models (GPT-2, DistilBERT). For larger models (Llama, Phi-4), a CUDA-capable GPU is recommended.

Verify Installation

garage --help

Or from Python:

import model_garage
print(model_garage.__version__)

Next Steps