| # ============================================================================ | |
| # PYTORCH CUDA WHEELS (CUDA 12.1) | |
| # ============================================================================ | |
| --extra-index-url https://download.pytorch.org/whl/cu121 | |
| # ============================================================================ | |
| # CORE PYTHON DEPENDENCIES (Python 3.10) | |
| # ============================================================================ | |
| numpy==1.26.4 | |
| Pillow>=10.0.1,<11.0 | |
| setuptools>=65.7.0,<69.0 | |
| wheel>=0.40.0,<1.0 | |
| typing-extensions>=4.12.2,<5.0 | |
| # ============================================================================ | |
| # WEB FRAMEWORK & UI | |
| # ============================================================================ | |
| gradio==4.44.1 | |
| gradio_client==1.3.0 | |
| # ============================================================================ | |
| # DEEP LEARNING & AI MODELS (Python 3.10 + CUDA 12.1) | |
| # ============================================================================ | |
| torch==2.5.1 | |
| torchvision==0.20.1 | |
| torchaudio==2.5.1 | |
| # Hugging Face ecosystem - UPDATED for SAM2 support | |
| transformers>=4.45.0 | |
| huggingface_hub>=0.25.0 | |
| accelerate>=0.20.3,<1.0 | |
| safetensors==0.4.3 | |
| tokenizers>=0.19.0 | |
| # Model utilities - MatAnyOne requirements | |
| einops==0.8.0 | |
| timm>=0.9.16 | |
| # ============================================================================ | |
| # COMPUTER VISION & VIDEO PROCESSING | |
| # ============================================================================ | |
| opencv-python-headless==4.10.0.84 | |
| # Video processing | |
| moviepy>=1.0.3,<2.0 | |
| imageio==2.25.0 | |
| imageio-ffmpeg>=0.4.8,<1.0 | |
| ffmpeg-python>=0.2.0,<1.0 | |
| # ============================================================================ | |
| # SCIENTIFIC COMPUTING | |
| # ============================================================================ | |
| scipy==1.13.1 | |
| tqdm>=4.66.1,<5.0 | |
| # ============================================================================ | |
| # CONFIGURATION & UTILITIES | |
| # ============================================================================ | |
| hydra-core==1.3.2 | |
| omegaconf==2.3.0 | |
| diskcache>=5.6.3,<6.0 | |
| psutil>=5.9.0,<6.0 | |
| # ============================================================================ | |
| # MATANYONE DEPENDENCIES (Python 3.10 compatible) | |
| # ============================================================================ | |
| easydict==1.10 | |
| gdown>=4.7.1 | |
| hickle>=5.0 | |
| cchardet>=2.1.7 | |
| gitpython>=3.1 | |
| netifaces>=0.11.0 | |
| pycocotools>=2.0.7 | |
| tensorboard>=2.11 | |
| # ============================================================================ | |
| # GIT DEPENDENCIES (Model Repositories) | |
| # ============================================================================ | |
| # SAM2 (Python 3.10+) - UPDATED to latest commit | |
| git+https://github.com/facebookresearch/segment-anything-2.git | |
| # MatAnyOne | |
| git+https://github.com/pq-yang/MatAnyone.git@2234ce5cdc487749515518bd035b5e18bccea3da |