MogensR commited on
Commit
261ae0e
·
1 Parent(s): 581de89

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -3
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
  # Core Dependencies (pinned for stability)
2
- numpy==1.24.3
3
  Pillow==10.0.1
4
  tqdm==4.66.1
5
 
@@ -10,7 +10,7 @@ gradio==4.44.1
10
  # Removed +cu118 suffix to let HF Spaces install the correct CUDA version
11
  torch>=2.0.1
12
  torchvision>=0.15.2
13
- torchaudio>=2.0.2
14
  transformers==4.30.2
15
  huggingface_hub==0.16.4
16
  accelerate==0.20.3
@@ -36,7 +36,8 @@ scipy==1.10.1
36
  typing-extensions==4.5.0 # For type hints
37
  setuptools==65.7.0 # For package installation
38
  wheel==0.40.0 # For wheel support
 
39
 
40
  # Git Dependencies (SAM2 and MatAnyone)
41
  git+https://github.com/facebookresearch/segment-anything-2.git@main
42
- git+https://github.com/pq-yang/MatAnyone.git@main
 
1
  # Core Dependencies (pinned for stability)
2
+ numpy>=1.24.3,<1.25
3
  Pillow==10.0.1
4
  tqdm==4.66.1
5
 
 
10
  # Removed +cu118 suffix to let HF Spaces install the correct CUDA version
11
  torch>=2.0.1
12
  torchvision>=0.15.2
13
+ torchaudio>=0.0.2
14
  transformers==4.30.2
15
  huggingface_hub==0.16.4
16
  accelerate==0.20.3
 
36
  typing-extensions==4.5.0 # For type hints
37
  setuptools==65.7.0 # For package installation
38
  wheel==0.40.0 # For wheel support
39
+ psutil>=5.9.0 # For memory monitoring
40
 
41
  # Git Dependencies (SAM2 and MatAnyone)
42
  git+https://github.com/facebookresearch/segment-anything-2.git@main
43
+ git+https://github.com/pq-yang/MatAnyone.git@main