| # ─── Core DL ───────────────────────────────────────────────────────── | |
| torch | |
| torchvision | |
| diffusers==0.32.2 | |
| accelerate>=1.1.1 # Required for diffusers; ZeroGPU-friendly when `device_map="cpu"` | |
| transformers>=4.46.2 | |
| numpy==1.26.0 | |
| einops | |
| pydantic>=2.10.3 | |
| protobuf | |
| sentencepiece>=0.2.0 | |
| # ─── Image / Video ─────────────────────────────────────────────────── | |
| opencv-python-headless>=4.10.0.84 | |
| imageio>=2.35.1 | |
| imageio-ffmpeg>=0.5.1 | |
| decord # ⚠ Optional, may fail on some CPUs — fallback to OpenCV if needed | |
| scipy | |
| scikit-image | |
| scikit-video>=1.1.11 | |
| moviepy==2.2.1 | |
| # ─── Specialized libs ──────────────────────────────────────────────── | |
| SwissArmyTransformer>=0.4.12 # ⚠ Heavy — make sure it doesn't try to use CUDA internally | |
| spandrel>=0.4.0 # If fails, consider making it optional | |
| gdown | |
| qwen_vl_utils # ⚠ If Qwen2.5-VL model used, not ZeroGPU-compatible — disable | |
| openai>=1.54.0 | |
| # ─── UI ────────────────────────────────────────────────────────────── | |
| gradio==4.29.0 # ✅ Stable, ZeroGPU-ready (as long as backend models are CPU-safe) | |