瀏覽代碼

New bitsandbytes (with latest GPU support) (#554)

* switch bitsandbytes v0.37 (latest GPU support and better autograd)

---------

Co-authored-by: Tim Dettmers <tim.dettmers@gmail.com>
justheuristic 2 年之前
父節點
當前提交
e6b6219cb9
共有 3 個文件被更改,包括 4 次插入4 次删除
  1. 1 1
      .github/workflows/run-benchmarks.yml
  2. 2 2
      .github/workflows/run-tests.yml
  3. 1 1
      setup.py

+ 1 - 1
.github/workflows/run-benchmarks.yml

@@ -28,7 +28,7 @@ jobs:
           pip install -r requirements-dev.txt
       - name: Build bitsandbytes
         run: |
-          pip install bitsandbytes==0.34.0
+          pip install bitsandbytes==0.37.0
       - name: Build hivemind
         run: |
           pip install .

+ 2 - 2
.github/workflows/run-tests.yml

@@ -31,7 +31,7 @@ jobs:
           pip install -r requirements-dev.txt
       - name: Build bitsandbytes
         run: |
-          pip install bitsandbytes==0.34.0
+          pip install bitsandbytes==0.37.0
       - name: Build hivemind
         run: |
           pip install .
@@ -93,7 +93,7 @@ jobs:
           pip install -r requirements-dev.txt
       - name: Build bitsandbytes
         run: |
-          pip install bitsandbytes==0.34.0
+          pip install bitsandbytes==0.37.0
       - name: Build hivemind
         run: |
           pip install -e . --no-use-pep517

+ 1 - 1
setup.py

@@ -139,7 +139,7 @@ with open("requirements-dev.txt") as dev_requirements_file:
 with open("requirements-docs.txt") as docs_requirements_file:
     extras["docs"] = list(map(str, parse_requirements(docs_requirements_file)))
 
-extras["bitsandbytes"] = ["bitsandbytes~=0.34.0"]
+extras["bitsandbytes"] = ["bitsandbytes~=0.37.0"]
 
 extras["all"] = extras["dev"] + extras["docs"] + extras["bitsandbytes"]