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
|