Pārlūkot izejas kodu

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.
Priyanshupareek 1 gadu atpakaļ
vecāks
revīzija
e268c99a6b
1 mainītis faili ar 1 papildinājumiem un 1 dzēšanām
  1. 1 1
      setup.cfg

+ 1 - 1
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