From a598387829aced4418a673fe31aab02e2f32bf8d Mon Sep 17 00:00:00 2001
From: dbogunowicz <97082108+dbogunowicz@users.noreply.github.com>
Date: Thu, 25 Apr 2024 17:13:29 +0200
Subject: [PATCH] Set the same ceiling for torch dep as in `sparseml`

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

diff --git a/setup.py b/setup.py
index dee52d96..ed36a638 100644
--- a/setup.py
+++ b/setup.py
@@ -25,7 +25,7 @@ def _setup_packages() -> List:
     )
     
 def _setup_install_requires() -> List:
-    return ["torch>=1.7.0", "transformers<4.41", "pydantic<2.7"]
+    return ["torch>=1.7.0,<2.3", "transformers<4.41", "pydantic<2.7"]
 
 def _setup_extras() -> Dict:
     return {"dev": ["black==22.12.0", "isort==5.8.0", "wheel>=0.36.2", "flake8>=3.8.3", "pytest>=6.0.0", "nbconvert>=7.16.3"]}