From e73283a8798d0665d9358273c073ddc2dcf6c3b2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Cl=C3=A9ment=20Chadebec?= Date: Sun, 22 Oct 2023 20:18:05 +0200 Subject: [PATCH] bump torch --- requirements.txt | 2 +- setup.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/requirements.txt b/requirements.txt index 6003534b..a0250f88 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,7 +4,7 @@ numpy>=1.19 pydantic>=2.0 scikit-learn scipy>=1.7.1 -torch>=2.0 +torch>=2.0.1 tqdm typing_extensions dataclasses>=0.6 diff --git a/setup.py b/setup.py index bdfe448c..3beb83de 100644 --- a/setup.py +++ b/setup.py @@ -32,7 +32,7 @@ "pydantic>=2.0", "scikit-learn", "scipy>=1.7.1", - "torch>=2.0", + "torch>=2.0.1", "tqdm", "typing_extensions", "dataclasses>=0.6",