AkCh12 commited on
Commit
c5ce01c
·
verified ·
1 Parent(s): e857b33

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -1
requirements.txt CHANGED
@@ -1,4 +1,6 @@
1
- torch
 
 
2
  transformers
3
  fastapi
4
  uvicorn
 
1
+ --extra-index-url https://download.pytorch.org/whl/cpu
2
+ torch==2.2.2+cpu
3
+ torchvision==0.17.2+cpu
4
  transformers
5
  fastapi
6
  uvicorn