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.).
This 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.Manual Installation
Requirements: Python 3.11+, Git, and uv. All dependencies (PyTorch, transformers,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
Once complete, you can run a demo for an end-to end example, or try the model on your data.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

