Spaces:
Running on Zero
Running on Zero
| # Base | |
| numpy<=1.26.4 | |
| scipy | |
| addict | |
| timm | |
| psutil | |
| huggingface_hub | |
| opencv-python-headless | |
| einops | |
| ninja | |
| torch==2.4.0 | |
| torchvision==0.19.0 | |
| torchaudio==2.4.0 | |
| # Extra dependencies | |
| --find-links https://data.pyg.org/whl/torch-2.4.0+cu121.html | |
| torch-scatter | |
| spconv-cu120 | |
| https://github.com/Dao-AILab/flash-attention/releases/download/v2.6.3/flash_attn-2.6.3+cu123torch2.4cxx11abiFALSE-cp310-cp310-linux_x86_64.whl | |
| # Visualization | |
| trimesh | |
| matplotlib |