Skip to content

Commit

Permalink
Restrict PyTorch version to <2.3.0 to resolve import error (#577)
Browse files Browse the repository at this point in the history
* 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.
  • Loading branch information
Priyanshupareek committed Apr 27, 2024
1 parent 30f522d commit e268c99
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion setup.cfg
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit e268c99

Please sign in to comment.