Ani07-05 commited on
Commit
60c44b1
·
1 Parent(s): 7b9bc7a

Add accelerate dependency

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -0
requirements.txt CHANGED
@@ -2,4 +2,5 @@ transformers>=4.30.0
2
  torch>=2.0.0
3
  streamlit>=1.20.0
4
  sentencepiece
 
5
  # accelerate # Optional, might help with model loading
 
2
  torch>=2.0.0
3
  streamlit>=1.20.0
4
  sentencepiece
5
+ accelerate # Make sure accelerate is installed
6
  # accelerate # Optional, might help with model loading