From d24d0bc763828bb21879e0633fefa00fd1699197 Mon Sep 17 00:00:00 2001 From: Rosie Wood Date: Thu, 24 Oct 2024 10:41:54 +0100 Subject: [PATCH] update setup.py and ci install --- .github/workflows/mr_ci_text_spotting.yml | 1 - setup.py | 4 ++-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/.github/workflows/mr_ci_text_spotting.yml b/.github/workflows/mr_ci_text_spotting.yml index 5638333d..622a90a0 100644 --- a/.github/workflows/mr_ci_text_spotting.yml +++ b/.github/workflows/mr_ci_text_spotting.yml @@ -33,7 +33,6 @@ jobs: - name: Install dependencies run: | python -m pip install wheel ninja - python -m pip install torch==2.2.2 torchvision==0.17.2 -f https://download.pytorch.org/whl/torch_stable.html python -m pip install ".[dev, text]" # all dependencies python -m pip install pytest-cov diff --git a/setup.py b/setup.py index ae803496..31d4ef48 100644 --- a/setup.py +++ b/setup.py @@ -38,8 +38,8 @@ "nest-asyncio>=1.5.1,<2.0.0", "scikit-image>=0.18.3", "scikit-learn>=1.0.1,<2.0.0", - "torch>=1.10.0", - "torchvision>=0.11.1,<0.17.3", + "torch>=1.10.0,=<2.2.2", + "torchvision>=0.11.1,=<0.17.2", "jupyter>=1.0.0,<2.0.0", "ipykernel>=6.5.1,<7.0.0", "ipywidgets>=8.0.0,<9.0.0",