Cap transformers <4.41 (#77)

*Issue #, if available:* #76

*Description of changes:* transformers 4.41 broke something for us, we
need to look into it deeper


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 2024-05-18 06:53:40 +02:00 committed by GitHub
parent 7a019b3432
commit 55166d3227
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -5,7 +5,7 @@ requires-python = ">=3.8"
license = { file = "LICENSE" }
dependencies = [
"torch~=2.1", # package was tested on 2.2
"transformers~=4.31",
"transformers~=4.31,<4.41",
"accelerate",
]