mirror of
https://github.com/amazon-science/chronos-forecasting
synced 2026-05-21 16:48:28 +00:00
This pull request introduces configurable attention backends to the Chronos-2 model, allowing users to select between eager, SDPA, and FlashAttention-2 implementations. --------- Co-authored-by: Oleksandr Shchur <oleks.shchur@gmail.com> Co-authored-by: Abdul Fatir <Abdulfatirs@gmail.com> |
||
|---|---|---|
| .. | ||
| configs | ||
| train.py | ||