heerjtdev commited on
Commit
ada6dc7
·
verified ·
1 Parent(s): 66473a8

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +7 -3
requirements.txt CHANGED
@@ -4,19 +4,23 @@ scipy==1.11.4
4
  pandas>=2.0.0,<3.0
5
  scikit-learn>=1.3.0,<1.6.0
6
 
 
 
 
7
  # Deep Learning - CPU ONLY
8
  --extra-index-url https://download.pytorch.org/whl/cpu
9
  torch==2.2.0+cpu
10
  torchvision==0.17.0+cpu
11
 
12
- # Web Framework
13
- gradio==4.44.0
14
-
15
  # NLP
16
  transformers==4.37.0
17
  tokenizers>=0.15.0,<0.20.0
18
  TorchCRF==1.1.0
19
 
 
 
 
 
20
  # Computer Vision
21
  ultralytics>=8.0.0,<8.1.0
22
  opencv-python-headless>=4.8.0,<5.0.0
 
4
  pandas>=2.0.0,<3.0
5
  scikit-learn>=1.3.0,<1.6.0
6
 
7
+ # Web Framework
8
+ gradio==4.44.0
9
+
10
  # Deep Learning - CPU ONLY
11
  --extra-index-url https://download.pytorch.org/whl/cpu
12
  torch==2.2.0+cpu
13
  torchvision==0.17.0+cpu
14
 
 
 
 
15
  # NLP
16
  transformers==4.37.0
17
  tokenizers>=0.15.0,<0.20.0
18
  TorchCRF==1.1.0
19
 
20
+ # Optimization - DOWNGRADE TO COMPATIBLE VERSION
21
+ optimum==1.14.1 # <--- This version works with PyTorch 2.2.0
22
+ onnxruntime==1.16.3
23
+
24
  # Computer Vision
25
  ultralytics>=8.0.0,<8.1.0
26
  opencv-python-headless>=4.8.0,<5.0.0