浏览代码

Fix bitsandbytes version

Alexander Borzunov 3 年之前
父节点
当前提交
872db56e0b
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      requirements.txt

+ 1 - 1
requirements.txt

@@ -1,7 +1,7 @@
 https://github.com/learning-at-home/hivemind/archive/dalle-v1.zip
 https://github.com/learning-at-home/dalle-pytorch/archive/weight-sharing.zip
 torchmetrics==0.6.2
-bitsandbytes-cuda111==0.26.0
+bitsandbytes-cuda111==0.26.0.post2
 transformers==4.12.2
 tokenizers==0.10.3
 datasets==1.14.0