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.
main
Priyanshupareek 2 weeks ago committed by GitHub
parent 30f522d1a0
commit e268c99a6b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -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

Loading…
Cancel
Save