Restrict PyTorch version to <2.3.0 to resolve import error (#577) · e268c99a6b - Gogs

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 year ago

parent

30f522d1a0

commit

e268c99a6b

1 changed files with 1 additions and 1 deletions

Split View Show Diff Stats

						
							+ 1
							
							- 1
						
setup.cfg
							 
								View File
							
				@@ -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