From a4854c5763f8c7cceefb3276f6cafe31cd741923 Mon Sep 17 00:00:00 2001 From: bdvllrs Date: Tue, 17 Dec 2024 10:02:39 +0000 Subject: [PATCH] specify custom version --- .github/workflows/tests.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml index 6621235..0f300a4 100644 --- a/.github/workflows/tests.yml +++ b/.github/workflows/tests.yml @@ -41,7 +41,7 @@ jobs: poetry source add torch-cpu --priority=explicit --no-interaction https://download.pytorch.org/whl/cpu torch_version=$(poetry show torch --no-ansi | grep 'version' | awk '{print $3}') echo "torch@$torch_version@@" - poetry add --lock --no-interaction --source torch-cpu "torch" + poetry add --lock --no-interaction --source torch-cpu "torch@2.0.1" echo "done" echo $(poetry show torchvision --no-ansi | grep 'version') torchvision_version=$(poetry show torchvision --no-ansi | grep 'version' | awk '{print $3}')