zhang4096 commited on
Commit
630adf9
·
verified ·
1 Parent(s): 82f3f78

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -2
requirements.txt CHANGED
@@ -1,2 +1,4 @@
1
- torch
2
- transformers
 
 
 
1
+ torch==2.2.1
2
+ torchvision==0.17.1
3
+ transformers==4.39.0
4
+ transformers-stream-generator==0.0.4