lolout1 commited on
Commit
6079807
·
1 Parent(s): bab2af7

testing new requirements for hugging face spaces compatibility

Browse files
Files changed (1) hide show
  1. requirements.txt +37 -16
requirements.txt CHANGED
@@ -1,22 +1,43 @@
1
- # Don't include torch/torchvision/torchaudio here since they're in Dockerfile
2
- # Don't include detectron2 since it's installed separately in Dockerfile
3
 
4
- # Core dependencies
5
- gradio
6
- numpy
7
- opencv-python-headless
8
- pillow
9
- matplotlib
 
 
 
 
 
 
 
 
 
 
 
 
10
  scipy
11
- timm
 
 
 
 
12
  einops
 
 
 
 
 
 
 
 
 
 
 
 
13
 
14
- # natten with correct torch version
15
  -f https://shi-labs.com/natten/wheels/cpu/torch2.0.0/index.html
16
  natten==0.14.6
17
- # Additional dependencies
18
- diffdist
19
- pytest
20
- ftfy
21
- iopath
22
- pycocotools
 
1
+ --extra-index-url https://download.pytorch.org/whl/cpu
 
2
 
3
+ # Core dependencies - PyTorch first
4
+ torch==2.0.1+cpu
5
+ torchvision==0.15.2+cpu
6
+ torchaudio==2.0.2+cpu
7
+
8
+ # CORRECT: Use detectron2 wheel for torch 2.0 CPU (not 1.10)
9
+ --find-links https://dl.fbaipublicfiles.com/detectron2/wheels/cpu/torch2.0/index.html
10
+ detectron2==0.6
11
+
12
+ # Rest of your dependencies
13
+ numpy>=1.23.0
14
+ Pillow>=8.3.2
15
+ opencv-python
16
+ git+https://github.com/cocodataset/panopticapi.git
17
+ git+https://github.com/mcordts/cityscapesScripts.git
18
+ imutils
19
+ cython
20
+ setuptools==59.5.0
21
  scipy
22
+ shapely
23
+ h5py
24
+ submitit
25
+ scikit-image
26
+ timm==0.4.12
27
  einops
28
+ icecream
29
+ wandb
30
+ ftfy
31
+ regex
32
+ inflect
33
+ diffdist
34
+ gdown
35
+ huggingface_hub
36
+ pytorch_lightning
37
+ tqdm
38
+ gradio
39
+ wget
40
 
41
+ # NATTEN
42
  -f https://shi-labs.com/natten/wheels/cpu/torch2.0.0/index.html
43
  natten==0.14.6