Try using torch nightly.

ludwig-ai · Jul 11, 2023 · 8529f04 · 8529f04
1 parent 60f1416
commit 8529f04
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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