Update requirements.txt
Browse files- requirements.txt +15 -8
requirements.txt
CHANGED
|
@@ -1,3 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
# ============================================================================
|
| 2 |
# CORE PYTHON DEPENDENCIES (Python 3.10)
|
| 3 |
# ============================================================================
|
|
@@ -16,18 +21,17 @@ gradio_client==1.3.0
|
|
| 16 |
# ============================================================================
|
| 17 |
# DEEP LEARNING & AI MODELS (Python 3.10 + CUDA 12.1)
|
| 18 |
# ============================================================================
|
| 19 |
-
# PyTorch ecosystem - CUDA 12.1 compatible
|
| 20 |
torch==2.1.0
|
| 21 |
torchvision==0.16.0
|
| 22 |
torchaudio==2.1.0
|
| 23 |
|
| 24 |
-
# Hugging Face ecosystem -
|
| 25 |
transformers==4.43.3
|
| 26 |
huggingface_hub==0.24.5
|
| 27 |
accelerate>=0.20.3,<1.0
|
| 28 |
safetensors==0.4.3
|
| 29 |
|
| 30 |
-
# Model utilities -
|
| 31 |
einops==0.8.0
|
| 32 |
timm>=0.9.16
|
| 33 |
|
|
@@ -59,7 +63,7 @@ psutil>=5.9.0,<6.0
|
|
| 59 |
# ============================================================================
|
| 60 |
# MATANYONE DEPENDENCIES (Python 3.10 compatible)
|
| 61 |
# ============================================================================
|
| 62 |
-
easydict==
|
| 63 |
gdown>=4.7.1
|
| 64 |
hickle>=5.0
|
| 65 |
cchardet>=2.1.7
|
|
@@ -68,14 +72,17 @@ netifaces>=0.11.0
|
|
| 68 |
pycocotools>=2.0.7
|
| 69 |
tensorboard>=2.11
|
| 70 |
|
|
|
|
|
|
|
|
|
|
| 71 |
# ============================================================================
|
| 72 |
# GIT DEPENDENCIES (Model Repositories)
|
| 73 |
# ============================================================================
|
| 74 |
-
# SAM2
|
| 75 |
git+https://github.com/facebookresearch/segment-anything-2.git@2b90b9f5ceec907a1c18123530e92e794ad901a4
|
| 76 |
|
| 77 |
-
#
|
| 78 |
git+https://github.com/pq-yang/MatAnyone.git@2234ce5cdc487749515518bd035b5e18bccea3da
|
| 79 |
|
| 80 |
-
# Thin Plate Spline (
|
| 81 |
-
git+https://github.com/cheind/py-thin-plate-spline@f6995795397118b7d0ac01aecd3f39ffbfad9dee
|
|
|
|
| 1 |
+
# ============================================================================
|
| 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 |
# ============================================================================
|
|
|
|
| 21 |
# ============================================================================
|
| 22 |
# DEEP LEARNING & AI MODELS (Python 3.10 + CUDA 12.1)
|
| 23 |
# ============================================================================
|
|
|
|
| 24 |
torch==2.1.0
|
| 25 |
torchvision==0.16.0
|
| 26 |
torchaudio==2.1.0
|
| 27 |
|
| 28 |
+
# Hugging Face ecosystem - MatAnyOne compatible versions
|
| 29 |
transformers==4.43.3
|
| 30 |
huggingface_hub==0.24.5
|
| 31 |
accelerate>=0.20.3,<1.0
|
| 32 |
safetensors==0.4.3
|
| 33 |
|
| 34 |
+
# Model utilities - MatAnyOne requirements
|
| 35 |
einops==0.8.0
|
| 36 |
timm>=0.9.16
|
| 37 |
|
|
|
|
| 63 |
# ============================================================================
|
| 64 |
# MATANYONE DEPENDENCIES (Python 3.10 compatible)
|
| 65 |
# ============================================================================
|
| 66 |
+
easydict==1.10
|
| 67 |
gdown>=4.7.1
|
| 68 |
hickle>=5.0
|
| 69 |
cchardet>=2.1.7
|
|
|
|
| 72 |
pycocotools>=2.0.7
|
| 73 |
tensorboard>=2.11
|
| 74 |
|
| 75 |
+
# (Valgfrit – hvis du møder protobuf-konflikter, tilføj linjen nedenfor)
|
| 76 |
+
# protobuf<4
|
| 77 |
+
|
| 78 |
# ============================================================================
|
| 79 |
# GIT DEPENDENCIES (Model Repositories)
|
| 80 |
# ============================================================================
|
| 81 |
+
# SAM2 (Python 3.10+)
|
| 82 |
git+https://github.com/facebookresearch/segment-anything-2.git@2b90b9f5ceec907a1c18123530e92e794ad901a4
|
| 83 |
|
| 84 |
+
# MatAnyOne
|
| 85 |
git+https://github.com/pq-yang/MatAnyone.git@2234ce5cdc487749515518bd035b5e18bccea3da
|
| 86 |
|
| 87 |
+
# Thin Plate Spline (MatAnyOne dependency)
|
| 88 |
+
git+https://github.com/cheind/py-thin-plate-spline@f6995795397118b7d0ac01aecd3f39ffbfad9dee
|