lolout1 commited on
Commit
7b70ae0
·
1 Parent(s): a4eeda0

requirements.txt changes

Browse files
Files changed (1) hide show
  1. requirements.txt +11 -7
requirements.txt CHANGED
@@ -1,20 +1,20 @@
1
-
2
-
3
  --extra-index-url https://download.pytorch.org/whl/cpu
4
 
 
5
  torch==2.0.1+cpu
6
  torchvision==0.15.2+cpu
7
  torchaudio==2.0.2+cpu
 
 
 
 
 
8
  numpy>=1.23.0
9
  Pillow>=8.3.2
10
  opencv-python
11
-
12
- git+https://github.com/facebookresearch/[email protected]
13
  git+https://github.com/cocodataset/panopticapi.git
14
  git+https://github.com/mcordts/cityscapesScripts.git
15
  imutils
16
- -f https://shi-labs.com/natten/wheels/cpu/torch2.0.0/index.html
17
- natten==0.14.6
18
  cython
19
  setuptools==59.5.0
20
  scipy
@@ -26,7 +26,7 @@ timm==0.4.12
26
  einops
27
  icecream
28
  wandb
29
- ftfy
30
  regex
31
  inflect
32
  diffdist
@@ -36,3 +36,7 @@ pytorch_lightning
36
  tqdm
37
  gradio
38
  wget
 
 
 
 
 
 
 
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
+ # CRITICAL: Use pre-built detectron2 wheel for torch 2.0 CPU
9
+ detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cpu/torch2.0/index.html
10
+
11
+ # Rest of your dependencies
12
  numpy>=1.23.0
13
  Pillow>=8.3.2
14
  opencv-python
 
 
15
  git+https://github.com/cocodataset/panopticapi.git
16
  git+https://github.com/mcordts/cityscapesScripts.git
17
  imutils
 
 
18
  cython
19
  setuptools==59.5.0
20
  scipy
 
26
  einops
27
  icecream
28
  wandb
29
+ ftfy
30
  regex
31
  inflect
32
  diffdist
 
36
  tqdm
37
  gradio
38
  wget
39
+
40
+ # NATTEN should be installed last
41
+ -f https://shi-labs.com/natten/wheels/cpu/torch2.0.0/index.html
42
+ natten==0.14.6