chronos-forecasting/scripts/training
Kashif Rasul ca9c3275a2
[chronos-2] add support for SDPA (#331)
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>
2025-10-22 14:02:09 +02:00
..
configs Add support for causal models (#113) 2024-06-13 17:37:04 +02:00
train.py [chronos-2] add support for SDPA (#331) 2025-10-22 14:02:09 +02:00