Quickstart (Recommended)
- Clone the quickstart repo with
demo.pyand all dependencies. - Install uv if not already present.
- Run
uv syncto install the exact locked dependency set (PyTorch, transformers,smb_utils, etc.).
Manual Installation
Manual Installation
Requirements: Python 3.11+, Git, and uv.All dependencies (PyTorch, transformers,
The quickstart uses uv for dependency management. If you prefer a different tool (conda, pip, etc.), refer to
pyproject.toml in the quickstart repo for the list of dependencies.smb_utils, etc.) are installed from the lockfile, ensuring a reproducible environment.Verify Your Installation
From thequickstart/ directory, verify that everything is working:Next Steps
Troubleshooting
CUDA Not Detected
Ensure NVIDIA drivers are up to date. Run
nvidia-smi to verify GPU is accessible.Out of Memory
Reduce memory use via
torch.float16 or quantization (see below).Model Access Denied
Some models may require authentication. Run
huggingface-cli login with your token.Slow Download
Model downloads can be large (several GB). Ensure stable connection and sufficient disk space.
Memory optimization
For large cohorts and/or limited GPU memory, use half-precision or quantization:- Float16
- 8-bit Quantization

