nickil commited on
Commit
5a5b359
1 Parent(s): 727f9bc

changing gpu to cpu onnx

Browse files
Files changed (2) hide show
  1. packages.txt +0 -1
  2. requirements.txt +1 -1
packages.txt DELETED
@@ -1 +0,0 @@
1
- nvidia-cuda-toolkit
 
 
requirements.txt CHANGED
@@ -8,5 +8,5 @@ torch==1.10.2
8
  pytorch-lightning==1.6.0
9
  transformers==4.17.0
10
  datasets==1.18.2
11
- onnxruntime-gpu==1.10.0
12
  loguru==0.6.0
 
8
  pytorch-lightning==1.6.0
9
  transformers==4.17.0
10
  datasets==1.18.2
11
+ onnxruntime==1.10.0
12
  loguru==0.6.0