Lord-Raven commited on
Commit
19acbf6
·
1 Parent(s): b0d2a02

Trying to use ONNX model.

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -0
requirements.txt CHANGED
@@ -1,6 +1,7 @@
1
  fastapi==0.88.0
2
  json5==0.9.10
3
  numpy==1.23.4
 
4
  torch==1.12.1
5
  torchvision==0.13.1
6
  transformers==4.44.0
 
1
  fastapi==0.88.0
2
  json5==0.9.10
3
  numpy==1.23.4
4
+ onnxruntime==1.18.1
5
  torch==1.12.1
6
  torchvision==0.13.1
7
  transformers==4.44.0