Lord-Raven commited on
Commit
4d6800e
·
1 Parent(s): 0cca822

Trying to use ONNX model.

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -2,7 +2,7 @@ fastapi==0.88.0
2
  json5==0.9.10
3
  numpy==1.23.4
4
  onnxruntime==1.18.1
5
- onnx_transformers
6
  torch==1.12.1
7
  torchvision==0.13.1
8
  transformers==4.44.0
 
2
  json5==0.9.10
3
  numpy==1.23.4
4
  onnxruntime==1.18.1
5
+ https://github.com/patil-suraj/onnx_transformers
6
  torch==1.12.1
7
  torchvision==0.13.1
8
  transformers==4.44.0