marcelbinz commited on
Commit
ec245b7
·
verified ·
1 Parent(s): bb856a6

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -2
requirements.txt CHANGED
@@ -1,3 +1,4 @@
1
  gradio==4.21.0
2
- torch==2.5.1
3
- transformers
 
 
1
  gradio==4.21.0
2
+ transformers==4.40.2 # ≥4.40 works fine with Llama 3.1
3
+ accelerate==0.30.0 # gives device_map="auto"
4
+ torch==2.5.1 # CPU wheel; fits the supported list