Soumik555 commited on
Commit
be2e445
·
1 Parent(s): 125934e
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.35.2
4
- torch==2.1.0+cpu
5
  tokenizers==0.15.0
6
  accelerate==0.24.1
7
  gradio==4.7.1
 
1
  fastapi==0.104.1
2
  uvicorn[standard]==0.24.0
3
  transformers==4.35.2
4
+ torch==2.1.0
5
  tokenizers==0.15.0
6
  accelerate==0.24.1
7
  gradio==4.7.1