فهرست منبع

build cpuonly bitsandbytes

dbaranchuk 3 سال پیش
والد
کامیت
4801d01cc9
1فایلهای تغییر یافته به همراه7 افزوده شده و 1 حذف شده
  1. 7 1
      .github/workflows/run-tests.yaml

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

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