Relax torch compatibility to <3 (#277)

*Issue #, if available:* Solves #273

*Description of changes:*


By submitting this pull request, I confirm that you can use, modify,
copy, and redistribute this contribution, under the terms of your
choice.
This commit is contained in:
Lorenzo Stella 2025-02-06 16:29:32 +01:00 committed by GitHub
parent 39515ff0fc
commit 75803f241f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -11,7 +11,7 @@ readme = "README.md"
license = { file = "LICENSE" }
requires-python = ">=3.9"
dependencies = [
"torch>=2.0,<2.6", # package was tested on 2.2
"torch>=2.0,<3", # package was tested on 2.2
"transformers>=4.30,<4.48",
"accelerate>=0.32,<1",
]