Эх сурвалжийг харах

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>
(cherry picked from commit e6b6219cb923d3a107ce8b112b459d1dceb6b595)
justheuristic 2 жил өмнө
parent
commit
be3e19904e

+ 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"]