From ae30cc9ed30045bcdb33dc94f54292c7db235276 Mon Sep 17 00:00:00 2001 From: Aryan Date: Fri, 18 Oct 2024 13:24:14 +0200 Subject: [PATCH 1/3] pin max torch version --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 89e18cab629b..3954a6f42974 100644 --- a/setup.py +++ b/setup.py @@ -130,7 +130,7 @@ "regex!=2019.12.17", "requests", "tensorboard", - "torch>=1.4", + "torch>=1.4,<=2.4.0", "torchvision", "transformers>=4.41.2", "urllib3<=2.0.0", From 8b16125cf220b19e9e5b4d1220ec3b674e8102c9 Mon Sep 17 00:00:00 2001 From: Aryan Date: Fri, 18 Oct 2024 13:24:57 +0200 Subject: [PATCH 2/3] update --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 3954a6f42974..1c17e6a521ad 100644 --- a/setup.py +++ b/setup.py @@ -130,7 +130,7 @@ "regex!=2019.12.17", "requests", "tensorboard", - "torch>=1.4,<=2.4.0", + "torch>=1.4,<=2.4.1", "torchvision", "transformers>=4.41.2", "urllib3<=2.0.0", From c056f234aee673028f730235e05d9b95e340fc0f Mon Sep 17 00:00:00 2001 From: Aryan Date: Fri, 18 Oct 2024 17:49:50 +0530 Subject: [PATCH 3/3] Update setup.py --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 1c17e6a521ad..7a8cc898e005 100644 --- a/setup.py +++ b/setup.py @@ -130,7 +130,7 @@ "regex!=2019.12.17", "requests", "tensorboard", - "torch>=1.4,<=2.4.1", + "torch>=1.4,<2.5.0", "torchvision", "transformers>=4.41.2", "urllib3<=2.0.0",