zhang4096 commited on
Commit
698e402
·
verified ·
1 Parent(s): 630adf9

Update requirements.txt

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