Jordi Catafal commited on
Commit
d51407f
·
1 Parent(s): 734683c

solving issue with torch

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -1,7 +1,7 @@
1
  fastapi==0.104.1
2
  uvicorn[standard]==0.24.0
3
  transformers==4.36.0
4
- torch==2.1.0+cpu --index-url https://download.pytorch.org/whl/cpu
5
  sentence-transformers==2.2.2
6
  numpy<2.0.0
7
  scikit-learn==1.3.2
 
1
  fastapi==0.104.1
2
  uvicorn[standard]==0.24.0
3
  transformers==4.36.0
4
+ torch==2.1.0
5
  sentence-transformers==2.2.2
6
  numpy<2.0.0
7
  scikit-learn==1.3.2