acecalisto3 commited on
Commit
6712f9a
1 Parent(s): 798ffe1

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -1
requirements.txt CHANGED
@@ -1,7 +1,8 @@
1
  transformers
 
2
  huggingface_hub
3
  torch
4
  tf-keras
5
  tensorflow
6
  flax
7
- accelerate
 
1
  transformers
2
+ transformers[torch]
3
  huggingface_hub
4
  torch
5
  tf-keras
6
  tensorflow
7
  flax
8
+ accelerate==0.21.0