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]