venbab commited on
Commit
1aff1bb
·
verified ·
1 Parent(s): cd74509

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -3
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
- # ---- Core Torch stack (keep if the GPU image supports it)
2
  torch==2.1.0
3
  torchvision==0.16.0
4
  torchaudio==2.1.0
@@ -27,10 +27,10 @@ einops==0.7.0
27
  omegaconf==2.3.0
28
  regex==2023.12.25
29
 
30
- # ---- ONNX (keep if needed)
31
  onnxruntime==1.16.2
32
 
33
- # ---- Detectron2-related deps used by apply_net
34
  cloudpickle==2.2.1
35
  fvcore==0.1.5.post20221221
36
  pycocotools==2.0.7
 
1
+ # ---- Core Torch stack
2
  torch==2.1.0
3
  torchvision==0.16.0
4
  torchaudio==2.1.0
 
27
  omegaconf==2.3.0
28
  regex==2023.12.25
29
 
30
+ # ---- ONNX (optional; keep if used)
31
  onnxruntime==1.16.2
32
 
33
+ # ---- Detectron2-adjacent deps used by apply_net
34
  cloudpickle==2.2.1
35
  fvcore==0.1.5.post20221221
36
  pycocotools==2.0.7