From 0d437de2cf610a147dd16be70d903a3ebbb5e4b1 Mon Sep 17 00:00:00 2001
From: Priyanshupareek <37779762+Priyanshupareek@users.noreply.github.com>
Date: Thu, 25 Apr 2024 16:31:06 +0530
Subject: [PATCH] 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
---
 setup.cfg | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.cfg b/setup.cfg
index dc0bd4e..53acabe 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==2.2.2
     bitsandbytes==0.41.1
     accelerate>=0.27.2
     huggingface-hub>=0.11.1,<1.0.0