Update requirements.txt
Browse files- requirements.txt +19 -10
requirements.txt
CHANGED
|
@@ -2,6 +2,7 @@
|
|
| 2 |
# PYTORCH CUDA WHEELS (CUDA 12.1)
|
| 3 |
# ============================================================================
|
| 4 |
--extra-index-url https://download.pytorch.org/whl/cu121
|
|
|
|
| 5 |
# ============================================================================
|
| 6 |
# CORE PYTHON DEPENDENCIES (Python 3.10)
|
| 7 |
# ============================================================================
|
|
@@ -10,39 +11,48 @@ Pillow>=10.0.1,<11.0
|
|
| 10 |
setuptools>=65.7.0,<69.0
|
| 11 |
wheel>=0.40.0,<1.0
|
| 12 |
typing-extensions>=4.12.2,<5.0
|
|
|
|
| 13 |
# ============================================================================
|
| 14 |
# WEB FRAMEWORK & UI
|
| 15 |
# ============================================================================
|
| 16 |
gradio==4.44.1
|
| 17 |
gradio_client==1.3.0
|
|
|
|
| 18 |
# ============================================================================
|
| 19 |
# DEEP LEARNING & AI MODELS (Python 3.10 + CUDA 12.1)
|
| 20 |
# ============================================================================
|
| 21 |
torch==2.5.1
|
| 22 |
torchvision==0.20.1
|
| 23 |
torchaudio==2.5.1
|
| 24 |
-
|
| 25 |
-
|
| 26 |
-
|
|
|
|
| 27 |
accelerate>=0.20.3,<1.0
|
| 28 |
safetensors==0.4.3
|
|
|
|
|
|
|
| 29 |
# Model utilities - MatAnyOne requirements
|
| 30 |
einops==0.8.0
|
| 31 |
timm>=0.9.16
|
|
|
|
| 32 |
# ============================================================================
|
| 33 |
# COMPUTER VISION & VIDEO PROCESSING
|
| 34 |
# ============================================================================
|
| 35 |
opencv-python-headless==4.10.0.84
|
|
|
|
| 36 |
# Video processing
|
| 37 |
moviepy>=1.0.3,<2.0
|
| 38 |
imageio==2.25.0
|
| 39 |
imageio-ffmpeg>=0.4.8,<1.0
|
| 40 |
ffmpeg-python>=0.2.0,<1.0
|
|
|
|
| 41 |
# ============================================================================
|
| 42 |
# SCIENTIFIC COMPUTING
|
| 43 |
# ============================================================================
|
| 44 |
scipy==1.13.1
|
| 45 |
tqdm>=4.66.1,<5.0
|
|
|
|
| 46 |
# ============================================================================
|
| 47 |
# CONFIGURATION & UTILITIES
|
| 48 |
# ============================================================================
|
|
@@ -50,6 +60,7 @@ hydra-core==1.3.2
|
|
| 50 |
omegaconf==2.3.0
|
| 51 |
diskcache>=5.6.3,<6.0
|
| 52 |
psutil>=5.9.0,<6.0
|
|
|
|
| 53 |
# ============================================================================
|
| 54 |
# MATANYONE DEPENDENCIES (Python 3.10 compatible)
|
| 55 |
# ============================================================================
|
|
@@ -61,14 +72,12 @@ gitpython>=3.1
|
|
| 61 |
netifaces>=0.11.0
|
| 62 |
pycocotools>=2.0.7
|
| 63 |
tensorboard>=2.11
|
| 64 |
-
|
| 65 |
-
# protobuf<4
|
| 66 |
# ============================================================================
|
| 67 |
# GIT DEPENDENCIES (Model Repositories)
|
| 68 |
# ============================================================================
|
| 69 |
-
# SAM2 (Python 3.10+)
|
| 70 |
-
git+https://github.com/facebookresearch/segment-anything-2.git
|
|
|
|
| 71 |
# MatAnyOne
|
| 72 |
-
git+https://github.com/pq-yang/MatAnyone.git@2234ce5cdc487749515518bd035b5e18bccea3da
|
| 73 |
-
# Thin Plate Spline - Installed automatically by MatAnyone
|
| 74 |
-
# git+https://github.com/cheind/py-thin-plate-spline@f6995795397118b7d0ac01aecd3f39ffbfad9dee
|
|
|
|
| 2 |
# PYTORCH CUDA WHEELS (CUDA 12.1)
|
| 3 |
# ============================================================================
|
| 4 |
--extra-index-url https://download.pytorch.org/whl/cu121
|
| 5 |
+
|
| 6 |
# ============================================================================
|
| 7 |
# CORE PYTHON DEPENDENCIES (Python 3.10)
|
| 8 |
# ============================================================================
|
|
|
|
| 11 |
setuptools>=65.7.0,<69.0
|
| 12 |
wheel>=0.40.0,<1.0
|
| 13 |
typing-extensions>=4.12.2,<5.0
|
| 14 |
+
|
| 15 |
# ============================================================================
|
| 16 |
# WEB FRAMEWORK & UI
|
| 17 |
# ============================================================================
|
| 18 |
gradio==4.44.1
|
| 19 |
gradio_client==1.3.0
|
| 20 |
+
|
| 21 |
# ============================================================================
|
| 22 |
# DEEP LEARNING & AI MODELS (Python 3.10 + CUDA 12.1)
|
| 23 |
# ============================================================================
|
| 24 |
torch==2.5.1
|
| 25 |
torchvision==0.20.1
|
| 26 |
torchaudio==2.5.1
|
| 27 |
+
|
| 28 |
+
# Hugging Face ecosystem - UPDATED for SAM2 support
|
| 29 |
+
transformers>=4.45.0
|
| 30 |
+
huggingface_hub>=0.25.0
|
| 31 |
accelerate>=0.20.3,<1.0
|
| 32 |
safetensors==0.4.3
|
| 33 |
+
tokenizers>=0.19.0
|
| 34 |
+
|
| 35 |
# Model utilities - MatAnyOne requirements
|
| 36 |
einops==0.8.0
|
| 37 |
timm>=0.9.16
|
| 38 |
+
|
| 39 |
# ============================================================================
|
| 40 |
# COMPUTER VISION & VIDEO PROCESSING
|
| 41 |
# ============================================================================
|
| 42 |
opencv-python-headless==4.10.0.84
|
| 43 |
+
|
| 44 |
# Video processing
|
| 45 |
moviepy>=1.0.3,<2.0
|
| 46 |
imageio==2.25.0
|
| 47 |
imageio-ffmpeg>=0.4.8,<1.0
|
| 48 |
ffmpeg-python>=0.2.0,<1.0
|
| 49 |
+
|
| 50 |
# ============================================================================
|
| 51 |
# SCIENTIFIC COMPUTING
|
| 52 |
# ============================================================================
|
| 53 |
scipy==1.13.1
|
| 54 |
tqdm>=4.66.1,<5.0
|
| 55 |
+
|
| 56 |
# ============================================================================
|
| 57 |
# CONFIGURATION & UTILITIES
|
| 58 |
# ============================================================================
|
|
|
|
| 60 |
omegaconf==2.3.0
|
| 61 |
diskcache>=5.6.3,<6.0
|
| 62 |
psutil>=5.9.0,<6.0
|
| 63 |
+
|
| 64 |
# ============================================================================
|
| 65 |
# MATANYONE DEPENDENCIES (Python 3.10 compatible)
|
| 66 |
# ============================================================================
|
|
|
|
| 72 |
netifaces>=0.11.0
|
| 73 |
pycocotools>=2.0.7
|
| 74 |
tensorboard>=2.11
|
| 75 |
+
|
|
|
|
| 76 |
# ============================================================================
|
| 77 |
# GIT DEPENDENCIES (Model Repositories)
|
| 78 |
# ============================================================================
|
| 79 |
+
# SAM2 (Python 3.10+) - UPDATED to latest commit
|
| 80 |
+
git+https://github.com/facebookresearch/segment-anything-2.git
|
| 81 |
+
|
| 82 |
# MatAnyOne
|
| 83 |
+
git+https://github.com/pq-yang/MatAnyone.git@2234ce5cdc487749515518bd035b5e18bccea3da
|
|
|
|
|
|