ffreemt commited on
Commit
212682a
1 Parent(s): f924882

Update bitsandbytes-cuda113 device_map={'': 0} rich, check nvidia-smi with subproces.run

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -1
requirements.txt CHANGED
@@ -18,7 +18,8 @@ safetensors
18
 
19
  # bitsandbytes
20
  # git+https://github.com/TimDettmers/bitsandbytes.git
21
- bitsandbytes==0.39.0
 
22
 
23
  transformers_stream_generator
24
  scipy
 
18
 
19
  # bitsandbytes
20
  # git+https://github.com/TimDettmers/bitsandbytes.git
21
+ # bitsandbytes==0.39.0
22
+ bitsandbytes-cuda113
23
 
24
  transformers_stream_generator
25
  scipy