Olivier-52 commited on
Commit
22bc910
·
1 Parent(s): 67af143

Update requirements.txt

Browse files

align training and predict libraries

Files changed (1) hide show
  1. requirements.txt +3 -3
requirements.txt CHANGED
@@ -1,8 +1,8 @@
1
  mlflow==2.21.3
2
  scikit-learn==1.4.2
3
- transformers>=4.40.0
4
- torch>=2.0.0
5
- tokenizers>=0.15.0
6
  requests>=2.31.0,<3
7
  fastapi
8
  uvicorn[standard]
 
1
  mlflow==2.21.3
2
  scikit-learn==1.4.2
3
+ transformers==4.38.0
4
+ torch==2.2.0
5
+ tokenizers>=0.15.2
6
  requests>=2.31.0,<3
7
  fastapi
8
  uvicorn[standard]