alrichardbollans commited on
Commit
8989d13
·
1 Parent(s): e492da7

Try to fix build failure

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. requirements.txt +4 -5
Dockerfile CHANGED
@@ -10,7 +10,7 @@ RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
10
 
11
  RUN pip install --no-cache-dir pip -U && pip install --no-cache-dir datasets "huggingface-hub>=0.19" "hf_xet>=1.0.0,<2.0.0" "hf-transfer>=0.1.4" "protobuf<4" "click<8.1" "pydantic~=1.0"
12
 
13
- RUN pip install --no-cache-dir git+https://github.com/facebookresearch/detectron2.git
14
 
15
  COPY . .
16
 
 
10
 
11
  RUN pip install --no-cache-dir pip -U && pip install --no-cache-dir datasets "huggingface-hub>=0.19" "hf_xet>=1.0.0,<2.0.0" "hf-transfer>=0.1.4" "protobuf<4" "click<8.1" "pydantic~=1.0"
12
 
13
+ RUN pip install --no-cache-dir git+https://github.com/facebookresearch/detectron2.git@v0.6
14
 
15
  COPY . .
16
 
requirements.txt CHANGED
@@ -1,13 +1,12 @@
1
  shapely
2
  matplotlib
3
  pillow
4
- shiny
5
  shinywidgets
6
  pandas
7
  opencv-python-headless
8
  pyyaml==5.1
9
- torch
10
- torchvision
11
  ninja
12
- supervision
13
- torchvision
 
1
  shapely
2
  matplotlib
3
  pillow
4
+ shiny==1.4.0
5
  shinywidgets
6
  pandas
7
  opencv-python-headless
8
  pyyaml==5.1
9
+ torch==2.6.0
10
+ torchvision==0.21.0
11
  ninja
12
+ supervision