yukee1992 commited on
Commit
29b7918
·
verified ·
1 Parent(s): 7fbfabe

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -1
requirements.txt CHANGED
@@ -1,3 +1,4 @@
1
  torch>=2.2.0
2
  transformers>=4.40.0
3
- gradio>=4.24.0
 
 
1
  torch>=2.2.0
2
  transformers>=4.40.0
3
+ gradio>=4.24.0
4
+ accelerate>=0.29.0 # Required for device mapping