From e268c99a6b53eb4ab30ac208c8bc149ba374a013 Mon Sep 17 00:00:00 2001
From: Priyanshupareek <37779762+Priyanshupareek@users.noreply.github.com>
Date: Sat, 27 Apr 2024 19:05:07 +0530
Subject: [PATCH] Restrict PyTorch version to <2.3.0 to resolve import error
 (#577)

* Pin PyTorch version to 2.2.2 to resolve import error

Addressing the import error encountered with PyTorch 2.3.0 as detailed in issue #576.
fixes #576

* Update setup.cfg

Modified the version constraint for PyTorch in setup.cfg to torch>=1.12,<2.3.0 to avoid the import errors introduced in version 2.3.0 while still supporting earlier compatible versions. This change follows feedback from @mryab to allow flexibility for users on different versions.
---
 setup.cfg | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.cfg b/setup.cfg
index dc0bd4e..3053b4e 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -32,7 +32,7 @@ package_dir =
 packages = find:
 python_requires = >=3.8
 install_requires =
-    torch>=1.12
+    torch>=1.12,<2.3.0
     bitsandbytes==0.41.1
     accelerate>=0.27.2
     huggingface-hub>=0.11.1,<1.0.0