abdullah-tariq commited on
Commit
ae0e56d
1 Parent(s): f03e3a7

Remove accelerate deps

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -4
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
- transformers
2
- torch
3
- accelerate
4
- gradio
 
1
+ gradio
2
+ transformers[torch]>="4.28.1,<5"
3
+ accelerate>="0.16.0,<1"
4
+ torch>="1.13.1,<2"