| 1 | # ============================================================================= |
| 2 | # Open-source DMD inference - Python dependencies |
| 3 | # |
| 4 | # These pins match the conda environment we developed and tested against |
| 5 | # (CUDA 12.8). For PyTorch family wheels with CUDA, install from the |
| 6 | # official PyTorch index, e.g.: |
| 7 | # |
| 8 | # pip install --index-url https://download.pytorch.org/whl/cu128 \ |
| 9 | # torch==2.8.0 torchvision==0.23.0 torchaudio==2.8.0 |
| 10 | # |
| 11 | # Then install the rest of the requirements: |
| 12 | # |
| 13 | # pip install -r requirements.txt |
| 14 | # |
| 15 | # If you prefer conda, use `environment.yml` instead. |
| 16 | # ============================================================================= |
| 17 | |
| 18 | # --- core deep-learning stack --- |
| 19 | torch==2.8.0 |
| 20 | torchvision==0.23.0 |
| 21 | torchaudio==2.8.0 |
| 22 | triton==3.4.0 |
| 23 | |
| 24 | # --- HF transformers (Gemma 3 text encoder) --- |
| 25 | transformers==4.57.6 |
| 26 | safetensors==0.6.2 |
| 27 | |
| 28 | # --- numerical / utilities --- |
| 29 | numpy>=2.2,<3 |
| 30 | scipy>=1.13 |
| 31 | einops>=0.8 |
| 32 | Pillow>=10 |
| 33 | tqdm>=4.66 |
| 34 | |
| 35 | # --- video / audio I/O --- |
| 36 | av>=14.0 |
| 37 |