From a36fb1d9d26682643b8ee3c9c08040dd30d67890 Mon Sep 17 00:00:00 2001
From: Alain Vaucher <avaucher@google.com>
Date: Tue, 27 Aug 2024 18:19:18 +0200
Subject: [PATCH] Relax version requirement for torch

---
 setup.cfg | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.cfg b/setup.cfg
index dc1bb4b..bda5b6f 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -31,7 +31,7 @@ install_requires =
     pandas>=1.0.3
     rxn-chem-utils>=1.0.3
     scipy>=1.4.1
-    torch>=1.5.0,<2.1
+    torch>=1.5.0,<2.5
     transformers>=4.0.0
 
 [options.package_data]