瀏覽代碼

update hivemind and bitsandbytes

dbaranchuk 3 年之前
父節點
當前提交
6e1696b839
共有 2 個文件被更改,包括 3 次插入3 次删除
  1. 1 0
      .github/workflows/run-tests.yaml
  2. 2 3
      requirements.txt

+ 1 - 0
.github/workflows/run-tests.yaml

@@ -26,6 +26,7 @@ jobs:
         run: |
           python -m pip install --upgrade pip
           pip install -r requirements.txt
+          pip install -i https://test.pypi.org/simple bitsandbytes-cuda113
       - name: Delete previous model, if exists
         run: |
           export HF_TAG=$(python -c "import os; print(os.environ.get('GITHUB_HEAD_REF') or os.environ.get('GITHUB_REF_NAME'))")

+ 2 - 3
requirements.txt

@@ -1,6 +1,5 @@
 torch==1.12.0
 accelerate==0.10.0
 huggingface-hub==0.7.0
-bitsandbytes-cuda113==0.26.0
-https://github.com/learning-at-home/hivemind/archive/28261470e44f2ae4157d08b563b4d2771f3a9549.zip
-https://github.com/huggingface/transformers/archive/6589e510fa4e6c442059de2fab84752535de9b23.zip
+https://github.com/learning-at-home/hivemind/archive/20b3b3d5f225ed525515a5383a008a8f9fad8173.zip
+https://github.com/huggingface/transformers/archive/6589e510fa4e6c442059de2fab84752535de9b23.zip