Update model link to Hugging Face (#353)

*Issue #, if available:*

*Description of changes:*


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:
Oleksandr Shchur 2025-10-30 18:08:57 +01:00 committed by GitHub
parent c60bd2ce4e
commit 9f8fef2116
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 13 additions and 10 deletions

View file

@ -36,7 +36,7 @@ jobs:
run: pip install ".[dev]" -f https://download.pytorch.org/whl/cpu/torch_stable.html run: pip install ".[dev]" -f https://download.pytorch.org/whl/cpu/torch_stable.html
- name: Run Eval Script for Chronos-2 - name: Run Eval Script for Chronos-2
run: python scripts/evaluation/evaluate.py chronos-2 ci/evaluate/backtest_config.yaml $CHRONOS_2_RESULTS_CSV --model-id=s3://autogluon/chronos-2 --device=cpu --torch-dtype=float32 run: python scripts/evaluation/evaluate.py chronos-2 ci/evaluate/backtest_config.yaml $CHRONOS_2_RESULTS_CSV --model-id=amazon/chronos-2 --device=cpu --torch-dtype=float32
- name: Print Chronos-2 CSV - name: Print Chronos-2 CSV
run: cat $CHRONOS_2_RESULTS_CSV run: cat $CHRONOS_2_RESULTS_CSV

View file

@ -19,7 +19,7 @@
## 🚀 News ## 🚀 News
- **20 Oct 2025**: 🚀 [Chronos-2](https://arxiv.org/abs/2510.15821) released. It offers _zero-shot_ support for univariate, multivariate, and covariate-informed forecasting tasks. Chronos-2 achieves the best performance on fev-bench, GIFT-Eval and Chronos Benchmark II amongst pretrained models. Check out [this notebook](notebooks/chronos-2-quickstart.ipynb) to get started with Chronos-2. - **20 Oct 2025**: 🚀 [Chronos-2](https://huggingface.co/amazon/chronos-2) released. It offers _zero-shot_ support for univariate, multivariate, and covariate-informed forecasting tasks. Chronos-2 achieves the best performance on fev-bench, GIFT-Eval and Chronos Benchmark II amongst pretrained models. Check out [this notebook](notebooks/chronos-2-quickstart.ipynb) to get started with Chronos-2.
- **14 Feb 2025**: 🚀 Chronos-Bolt is now available on Amazon SageMaker JumpStart! Check out the [tutorial notebook](notebooks/deploy-chronos-bolt-to-amazon-sagemaker.ipynb) to learn how to deploy Chronos endpoints for production use in 3 lines of code. - **14 Feb 2025**: 🚀 Chronos-Bolt is now available on Amazon SageMaker JumpStart! Check out the [tutorial notebook](notebooks/deploy-chronos-bolt-to-amazon-sagemaker.ipynb) to learn how to deploy Chronos endpoints for production use in 3 lines of code.
- **12 Dec 2024**: 📊 We released [`fev`](https://github.com/autogluon/fev), a lightweight package for benchmarking time series forecasting models based on the [Hugging Face `datasets`](https://huggingface.co/docs/datasets/en/index) library. - **12 Dec 2024**: 📊 We released [`fev`](https://github.com/autogluon/fev), a lightweight package for benchmarking time series forecasting models based on the [Hugging Face `datasets`](https://huggingface.co/docs/datasets/en/index) library.
- **26 Nov 2024**: ⚡️ Chronos-Bolt models released [on HuggingFace](https://huggingface.co/collections/amazon/chronos-models-65f1791d630a8d57cb718444). Chronos-Bolt models are more accurate (5% lower error), up to 250x faster and 20x more memory efficient than the original Chronos models of the same size! - **26 Nov 2024**: ⚡️ Chronos-Bolt models released [on HuggingFace](https://huggingface.co/collections/amazon/chronos-models-65f1791d630a8d57cb718444). Chronos-Bolt models are more accurate (5% lower error), up to 250x faster and 20x more memory efficient than the original Chronos models of the same size!
@ -39,7 +39,7 @@ This package provides an interface to the Chronos family of **pretrained time se
| Model ID | Parameters | | Model ID | Parameters |
| ---------------------------------------------------------------------- | ---------- | | ---------------------------------------------------------------------- | ---------- |
| [`s3://autogluon/chronos-2`](https://arxiv.org/abs/2510.15821) | 120M | | [`amazon/chronos-2`](https://huggingface.co/amazon/chronos-2) | 120M |
| [`amazon/chronos-bolt-tiny`](https://huggingface.co/amazon/chronos-bolt-tiny) | 9M | | [`amazon/chronos-bolt-tiny`](https://huggingface.co/amazon/chronos-bolt-tiny) | 9M |
| [`amazon/chronos-bolt-mini`](https://huggingface.co/amazon/chronos-bolt-mini) | 21M | | [`amazon/chronos-bolt-mini`](https://huggingface.co/amazon/chronos-bolt-mini) | 21M |
| [`amazon/chronos-bolt-small`](https://huggingface.co/amazon/chronos-bolt-small) | 48M | | [`amazon/chronos-bolt-small`](https://huggingface.co/amazon/chronos-bolt-small) | 48M |
@ -68,7 +68,7 @@ A minimal example showing how to perform forecasting using Chronos-2:
import pandas as pd # requires: pip install 'pandas[pyarrow]' import pandas as pd # requires: pip install 'pandas[pyarrow]'
from chronos import Chronos2Pipeline from chronos import Chronos2Pipeline
pipeline = Chronos2Pipeline.from_pretrained("s3://autogluon/chronos-2", device_map="cuda") pipeline = Chronos2Pipeline.from_pretrained("amazon/chronos-2", device_map="cuda")
# Load historical target values and past values of covariates # Load historical target values and past values of covariates
context_df = pd.read_parquet("https://autogluon.s3.amazonaws.com/datasets/timeseries/electricity_price/train.parquet") context_df = pd.read_parquet("https://autogluon.s3.amazonaws.com/datasets/timeseries/electricity_price/train.parquet")

View file

@ -36,7 +36,7 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 1, "execution_count": null,
"id": "fcc7e496", "id": "fcc7e496",
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
@ -53,7 +53,7 @@
"\n", "\n",
"# Load the Chronos-2 pipeline\n", "# Load the Chronos-2 pipeline\n",
"# GPU recommended for faster inference, but CPU is also supported\n", "# GPU recommended for faster inference, but CPU is also supported\n",
"pipeline: Chronos2Pipeline = BaseChronosPipeline.from_pretrained(\"s3://autogluon/chronos-2/\", device_map=\"cuda\")" "pipeline: Chronos2Pipeline = BaseChronosPipeline.from_pretrained(\"amazon/chronos-2\", device_map=\"cuda\")"
] ]
}, },
{ {

View file

@ -13,10 +13,9 @@ import pytest
import torch import torch
from chronos import BaseChronosPipeline, Chronos2Pipeline from chronos import BaseChronosPipeline, Chronos2Pipeline
from chronos.chronos2.dataset import convert_df_input_to_list_of_dicts_input
from chronos.chronos2.config import Chronos2CoreConfig from chronos.chronos2.config import Chronos2CoreConfig
from chronos.chronos2.dataset import convert_df_input_to_list_of_dicts_input
from chronos.chronos2.layers import MHA from chronos.chronos2.layers import MHA
from test.util import validate_tensor from test.util import validate_tensor
DUMMY_MODEL_PATH = Path(__file__).parent / "dummy-chronos2-model" DUMMY_MODEL_PATH = Path(__file__).parent / "dummy-chronos2-model"
@ -35,6 +34,10 @@ def test_base_chronos2_pipeline_loads_from_s3():
BaseChronosPipeline.from_pretrained("s3://autogluon/chronos-2", device_map="cpu") BaseChronosPipeline.from_pretrained("s3://autogluon/chronos-2", device_map="cpu")
def test_base_chronos2_pipeline_loads_from_hf():
BaseChronosPipeline.from_pretrained("amazon/chronos-2", device_map="cpu")
@pytest.mark.parametrize( @pytest.mark.parametrize(
"inputs, prediction_length, expected_output_shapes", "inputs, prediction_length, expected_output_shapes",
[ [