testing new requirements for hugging face spaces compatibility
Browse files- requirements.txt +37 -16
requirements.txt
CHANGED
@@ -1,22 +1,43 @@
|
|
1 |
-
|
2 |
-
# Don't include detectron2 since it's installed separately in Dockerfile
|
3 |
|
4 |
-
# Core dependencies
|
5 |
-
|
6 |
-
|
7 |
-
|
8 |
-
|
9 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
10 |
scipy
|
11 |
-
|
|
|
|
|
|
|
|
|
12 |
einops
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
13 |
|
14 |
-
#
|
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
|
|
|
|
|
|
|
|
|
|
|
|