diff --git a/.github/workflows/notebook_tests.yml b/.github/workflows/notebook_tests.yml index 9c36c0152..087c03691 100644 --- a/.github/workflows/notebook_tests.yml +++ b/.github/workflows/notebook_tests.yml @@ -56,7 +56,7 @@ jobs: - name: GPU pip installs run: | pip install accelerate - CMAKE_ARGS="-DLLAMA_CUBLAS=on" pip install "llama-cpp-python<0.2.58" + CMAKE_ARGS="-DLLAMA_CUBLAS=on" pip install "llama-cpp-python!=0.2.58,!=0.2.75" - name: Check GPU available run: | python -c "import torch; assert torch.cuda.is_available()"