From 8529f04134e8fb768024ebef38eb8d017adf4d22 Mon Sep 17 00:00:00 2001 From: Justin Zhao Date: Tue, 11 Jul 2023 16:26:55 +0000 Subject: [PATCH] Try using torch nightly. --- .github/workflows/pytest.yml | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/.github/workflows/pytest.yml b/.github/workflows/pytest.yml index 3f135e1c150..94e8ad43843 100644 --- a/.github/workflows/pytest.yml +++ b/.github/workflows/pytest.yml @@ -121,8 +121,7 @@ jobs: if [ "$PYTORCH" == "nightly" ]; then extra_index_url=https://download.pytorch.org/whl/nightly/cpu - # TODO: https://github.com/ludwig-ai/ludwig/issues/3395 - pip install --pre torch==2.1.0.dev20230508+cpu torchtext torchvision torchaudio --index-url $extra_index_url + pip install --pre torch torchtext torchvision torchaudio --index-url $extra_index_url else extra_index_url=https://download.pytorch.org/whl/cpu pip install torch==$PYTORCH torchtext torchvision torchaudio --extra-index-url $extra_index_url