MogensR commited on
Commit
459d6d2
·
1 Parent(s): 31653b7

Update requirements.txt

Browse files
Files changed (1) hide show
  1. 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
- # Hugging Face ecosystem - MatAnyOne compatible versions
25
- transformers==4.43.3
26
- huggingface_hub==0.24.5
 
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
- # (Valgfrit – hvis du møder protobuf-konflikter, tilføj linjen nedenfor)
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@2b90b9f5ceec907a1c18123530e92e794ad901a4
 
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