mirror of
https://github.com/amazon-science/chronos-forecasting
synced 2026-05-24 10:08:33 +00:00
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:
parent
7a019b3432
commit
55166d3227
1 changed files with 1 additions and 1 deletions
|
|
@ -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",
|
||||
]
|
||||
|
||||
|
|
|
|||
Loading…
Reference in a new issue