Manju017 commited on
Commit
711f2c0
·
verified ·
1 Parent(s): b0ea0b7

Update bitsandbytes configuration

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
  transformers
2
  torch
3
  gradio
4
- bitsandbytes
5
  accelerate>=0.26.0
 
1
  transformers
2
  torch
3
  gradio
4
+ bitsandbytes[cuda]
5
  accelerate>=0.26.0