sproducts commited on
Commit
42c0780
·
verified ·
1 Parent(s): 204b284

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -2,5 +2,5 @@ gradio
2
  PyYAML
3
  transformers
4
  torch
5
- accelerate
6
- bitsandbytes-cuda117 # Or bitsandbytes-cuda118, try different CUDA versions
 
2
  PyYAML
3
  transformers
4
  torch
5
+ accelerate>=0.26.0
6
+ bitsandbytes==0.41.1 # Pinning to version 0.41.1