Spaces:
Build error
Build error
changing gpu to cpu onnx
Browse files- packages.txt +0 -1
- 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
|
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
|