CI: use CPU wheel repo for PyTorch

This commit is contained in:
Aarni Koskela 2023-05-11 10:36:10 +03:00
parent dd3ca9adf7
commit c702010e57
1 changed files with 5 additions and 0 deletions

View File

@ -19,6 +19,11 @@ jobs:
**/requirements*txt **/requirements*txt
- name: Run tests - name: Run tests
run: python launch.py --tests test --no-half --disable-opt-split-attention --use-cpu all --skip-torch-cuda-test run: python launch.py --tests test --no-half --disable-opt-split-attention --use-cpu all --skip-torch-cuda-test
env:
PIP_DISABLE_PIP_VERSION_CHECK: "1"
PIP_PROGRESS_BAR: "off"
TORCH_INDEX_URL: https://download.pytorch.org/whl/cpu
WEBUI_LAUNCH_LIVE_OUTPUT: "1"
- name: Upload main app stdout-stderr - name: Upload main app stdout-stderr
uses: actions/upload-artifact@v3 uses: actions/upload-artifact@v3
if: always() if: always()