Selaa lähdekoodia

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 vuotta sitten
vanhempi
commit
be3e19904e
3 muutettua tiedostoa jossa 4 lisäystä ja 4 poistoa
  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"]