Spaces:
Running on Zero
Running on Zero
| --extra-index-url https://download.pytorch.org/whl/cu124 | |
| # Core model runtime | |
| torch==2.5.1 | |
| torchvision==0.20.1 | |
| transformers==4.51.3 | |
| tokenizers==0.21.4 | |
| accelerate==1.10.1 | |
| huggingface_hub==0.34.4 | |
| sentencepiece==0.1.99 | |
| timm==1.0.3 | |
| numpy>=1.26.0 | |
| Pillow | |
| einops==0.6.1 | |
| einops-exts==0.0.4 | |
| # Image and video processing | |
| decord==0.6.0 | |
| imageio==2.34.0 | |
| imageio-ffmpeg==0.4.9 | |
| opencv-python-headless | |
| ffmpeg-python | |
| requests | |
| # UI | |
| gradio>=5.44.1,<7 | |
| # FlashAttention — pre-built wheel from GitHub (no nvcc required) | |
| https://github.com/Dao-AILab/flash-attention/releases/download/v2.8.3/flash_attn-2.8.3+cu12torch2.5cxx11abiFALSE-cp312-cp312-linux_x86_64.whl | |