diff --git a/requirements.txt b/requirements.txt index 8ee7b62..c2c6a84 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,6 @@ ---extra-index-url https://download.pytorch.org/whl/cu121 +# --extra-index-url https://download.pytorch.org/whl/cu121 numpy==1.25.2 -torch==2.1.1+cu121 +# torch==2.1.1+cu121 torchvision torchaudio albumentations