diff --git a/requirements/torch.txt b/requirements/torch.txt index 5f6aa46a5..f123d047e 100644 --- a/requirements/torch.txt +++ b/requirements/torch.txt @@ -1,4 +1,4 @@ ---index-url https://download.pytorch.org/whl/cu121 +--extra-index-url https://download.pytorch.org/whl/cu121 torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0