Explorar o código

build cpuonly bitsandbytes

dbaranchuk %!s(int64=3) %!d(string=hai) anos
pai
achega
4801d01cc9
Modificáronse 1 ficheiros con 7 adicións e 1 borrados
  1. 7 1
      .github/workflows/run-tests.yaml

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

@@ -26,7 +26,6 @@ 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'))")
@@ -63,6 +62,13 @@ jobs:
           python -m pip install --upgrade pip
           pip install -r requirements.txt
           pip install -r requirements-dev.txt
+      - name: Build bitsandbytes cpuonly
+        run: |
+          git clone https://github.com/TimDettmers/bitsandbytes.git
+          cd bitsandbytes
+          make cpuonly
+          pip install .
+          cd -
       - name: Test
         run: |
           export HF_TAG=$(python -c "import os; print(os.environ.get('GITHUB_HEAD_REF') or os.environ.get('GITHUB_REF_NAME'))")