Fix BNB OOM by pinning version

2023-05-09 02:10:31 +09:00
parent 7576d85c73
commit fe582df7d3
1 changed files with 1 additions and 1 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,7 +4,7 @@ attrdict
 fire
 PyYAML==6.0
 black
-bitsandbytes
+bitsandbytes==0.37.2
 datasets
 accelerate
 sentencepiece