rogier1970 commited on
Commit
a09f92d
·
verified ·
1 Parent(s): 122b046

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +18 -11
requirements.txt CHANGED
@@ -1,13 +1,22 @@
1
- openmim
2
- mmcv-lite>=2.0.0rc4,<2.1.0
3
- mmdet>=3.0.0
4
- mmengine>=0.7.1
5
- mmyolo
 
 
 
 
 
 
 
 
 
 
 
6
  gradio
7
- transformers
8
  addict
9
  yapf
10
- numpy
11
  opencv-python
12
  supervision==0.18.0
13
  ftfy
@@ -20,7 +29,5 @@ onnxruntime
20
  onnxruntime-gpu
21
  onnx-simplifier
22
 
23
- --extra-index-url https://download.pytorch.org/whl/cu118
24
- torch==2.0.1
25
- torchvision==0.15.2
26
- fastapi==0.111.0
 
1
+ # NumPy 2.x geeft ABI-problemen met binaire wheels in deze stack
2
+ numpy==1.26.4
3
+
4
+ # CUDA 11.8 + Torch
5
+ --extra-index-url https://download.pytorch.org/whl/cu118
6
+ torch==2.0.1+cu118
7
+ torchvision==0.15.2+cu118
8
+
9
+ # MMCV met compiled ops (mmcv._ext); let op: dit werkt alleen als je Space GPU/CUDA gebruikt
10
+ -f https://download.openmmlab.com/mmcv/dist/cu118/torch2.0.0/index.html
11
+ mmcv==2.0.1
12
+
13
+ mmengine==0.10.7
14
+ mmdet==3.3.0
15
+ mmyolo==0.6.0
16
+
17
  gradio
 
18
  addict
19
  yapf
 
20
  opencv-python
21
  supervision==0.18.0
22
  ftfy
 
29
  onnxruntime-gpu
30
  onnx-simplifier
31
 
32
+ # Belangrijk: voorkom dat een recente transformers-versie Torch>=2.1 afdwingt
33
+ transformers==4.35.2