Elgato_dark/chronos-forecasting
1
0
Fork 0
mirror of https://github.com/amazon-science/chronos-forecasting synced 2026-05-24 10:08:33 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
chronos-forecasting/test
History
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
..
dummy-chronos-bolt-model ⚡ Add support for Chronos-Bolt models (#204) 2024-11-26 17:47:14 +01:00
dummy-chronos-model Upload code 2024-03-13 09:58:39 +01:00
dummy-chronos2-model Add Chronos-2 (#319) 2025-10-20 10:34:20 +02:00
__init__.py Fix padding for int contexts (#227) 2024-12-04 16:46:17 +01:00
test_chronos.py Add Chronos-2 (#319) 2025-10-20 10:34:20 +02:00
test_chronos2.py [chronos-2] add support for SDPA (#331) 2025-10-22 14:02:09 +02:00
test_chronos_bolt.py Add Chronos-2 (#319) 2025-10-20 10:34:20 +02:00
test_utils.py Add Chronos-2 (#319) 2025-10-20 10:34:20 +02:00
util.py Add Chronos-2 (#319) 2025-10-20 10:34:20 +02:00