MogensR commited on
Commit
c0056b7
·
verified ·
1 Parent(s): 1073095

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +14 -14
requirements.txt CHANGED
@@ -1,11 +1,11 @@
1
- # Core dependencies
2
  gradio==4.44.1
3
  numpy==1.24.3
4
  opencv-python-headless==4.8.1.78
5
  Pillow==10.1.0
6
  moviepy==1.0.3
7
 
8
- # PyTorch GPU version for HuggingFace Pro
9
  --extra-index-url https://download.pytorch.org/whl/cu121
10
  torch==2.2.0
11
  torchvision==0.17.0
@@ -13,32 +13,32 @@ torchvision==0.17.0
13
  # Background removal
14
  rembg==2.0.50
15
 
16
- # SAM2 dependencies
17
  matplotlib==3.7.2
18
- onnxruntime-gpu==1.16.0
19
- scikit-image==0.21.0
20
- hydra-core>=1.3.2
21
- omegaconf>=2.3.0
22
- iopath>=0.1.10
23
  hydra-core>=1.3.2
24
  omegaconf>=2.3.0
25
  iopath>=0.1.10
26
  submitit>=1.4.5
 
27
 
28
- # MatAnyone dependencies
29
  scipy==1.11.1
30
  tqdm==4.66.1
31
  kornia>=0.7.0
32
  einops>=0.7.0
 
33
 
34
- # Additional utilities
35
  requests==2.31.0
36
  filelock==3.12.2
37
- huggingface-hub>=0.19.3
38
 
39
- # Video processing
40
  imageio==2.31.1
41
  imageio-ffmpeg==0.4.9
42
 
43
- # CUDA optimization
44
- nvidia-ml-py3==7.352.0
 
 
 
 
1
+ # Core
2
  gradio==4.44.1
3
  numpy==1.24.3
4
  opencv-python-headless==4.8.1.78
5
  Pillow==10.1.0
6
  moviepy==1.0.3
7
 
8
+ # PyTorch (CUDA 12.1 wheels)
9
  --extra-index-url https://download.pytorch.org/whl/cu121
10
  torch==2.2.0
11
  torchvision==0.17.0
 
13
  # Background removal
14
  rembg==2.0.50
15
 
16
+ # SAM2 + config stack
17
  matplotlib==3.7.2
 
 
 
 
 
18
  hydra-core>=1.3.2
19
  omegaconf>=2.3.0
20
  iopath>=0.1.10
21
  submitit>=1.4.5
22
+ scikit-image==0.21.0
23
 
24
+ # MatAnyone deps (add timm — commonly required)
25
  scipy==1.11.1
26
  tqdm==4.66.1
27
  kornia>=0.7.0
28
  einops>=0.7.0
29
+ timm>=0.9.12 # <-- add this
30
 
31
+ # HF + utilities
32
  requests==2.31.0
33
  filelock==3.12.2
34
+ huggingface-hub>=0.22.0 # bump slightly; more reliable HF downloads
35
 
36
+ # Video IO
37
  imageio==2.31.1
38
  imageio-ffmpeg==0.4.9
39
 
40
+ # GPU monitoring
41
+ nvidia-ml-py3==7.352.0
42
+
43
+ # Optional (only if MatAnyone ever complains about it; otherwise omit)
44
+ # decord==0.6.0