Commit
·
5108440
1
Parent(s):
de2ceb8
Add model, update README
Browse files- README.md +38 -3
- config.json +50 -0
- model.safetensors +3 -0
README.md
CHANGED
|
@@ -1,3 +1,38 @@
|
|
| 1 |
-
---
|
| 2 |
-
license: apache-2.0
|
| 3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: apache-2.0
|
| 3 |
+
model_id: chronos-2-small
|
| 4 |
+
tags:
|
| 5 |
+
- time series
|
| 6 |
+
- forecasting
|
| 7 |
+
- foundation models
|
| 8 |
+
- pretrained models
|
| 9 |
+
- safetensors
|
| 10 |
+
paper:
|
| 11 |
+
- https://arxiv.org/abs/2510.15821
|
| 12 |
+
datasets:
|
| 13 |
+
- autogluon/chronos_datasets
|
| 14 |
+
- Salesforce/GiftEvalPretrain
|
| 15 |
+
leaderboards:
|
| 16 |
+
- Salesforce/GIFT-Eval
|
| 17 |
+
- autogluon/fev-leaderboard
|
| 18 |
+
pipeline_tag: time-series-forecasting
|
| 19 |
+
library_name: chronos-forecasting
|
| 20 |
+
|
| 21 |
+
---
|
| 22 |
+
|
| 23 |
+
# Chronos-2-Small
|
| 24 |
+
|
| 25 |
+
This is the _small_ variant of the Chronos-2 model with 28M parameters. For details the Chronos-2 model, please refer to https://huggingface.co/autogluon/chronos-2.
|
| 26 |
+
|
| 27 |
+
## Citation
|
| 28 |
+
|
| 29 |
+
If you find Chronos-2 useful for your research, please consider citing the associated paper:
|
| 30 |
+
|
| 31 |
+
```
|
| 32 |
+
@article{ansari2025chronos2,
|
| 33 |
+
title = {Chronos-2: From Univariate to Universal Forecasting},
|
| 34 |
+
author = {Abdul Fatir Ansari and Oleksandr Shchur and Jaris Küken and Andreas Auer and Boran Han and Pedro Mercado and Syama Sundar Rangapuram and Huibin Shen and Lorenzo Stella and Xiyuan Zhang and Mononito Goswami and Shubham Kapoor and Danielle C. Maddix and Pablo Guerron and Tony Hu and Junming Yin and Nick Erickson and Prateek Mutalik Desai and Hao Wang and Huzefa Rangwala and George Karypis and Yuyang Wang and Michael Bohlke-Schneider},
|
| 35 |
+
year = {2025},
|
| 36 |
+
url = {https://arxiv.org/abs/2510.15821}
|
| 37 |
+
}
|
| 38 |
+
```
|
config.json
ADDED
|
@@ -0,0 +1,50 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"architectures": [
|
| 3 |
+
"Chronos2Model"
|
| 4 |
+
],
|
| 5 |
+
"chronos_config": {
|
| 6 |
+
"context_length": 8192,
|
| 7 |
+
"input_patch_size": 16,
|
| 8 |
+
"input_patch_stride": 16,
|
| 9 |
+
"max_output_patches": 64,
|
| 10 |
+
"output_patch_size": 16,
|
| 11 |
+
"quantiles": [
|
| 12 |
+
0.01,
|
| 13 |
+
0.05,
|
| 14 |
+
0.1,
|
| 15 |
+
0.2,
|
| 16 |
+
0.3,
|
| 17 |
+
0.4,
|
| 18 |
+
0.5,
|
| 19 |
+
0.6,
|
| 20 |
+
0.7,
|
| 21 |
+
0.8,
|
| 22 |
+
0.9,
|
| 23 |
+
0.95,
|
| 24 |
+
0.99
|
| 25 |
+
],
|
| 26 |
+
"time_encoding_scale": 8192,
|
| 27 |
+
"use_arcsinh": true,
|
| 28 |
+
"use_reg_token": true
|
| 29 |
+
},
|
| 30 |
+
"chronos_pipeline_class": "Chronos2Pipeline",
|
| 31 |
+
"d_ff": 2048,
|
| 32 |
+
"d_kv": 64,
|
| 33 |
+
"d_model": 512,
|
| 34 |
+
"dense_act_fn": "relu",
|
| 35 |
+
"dropout_rate": 0.1,
|
| 36 |
+
"feed_forward_proj": "relu",
|
| 37 |
+
"initializer_factor": 0.05,
|
| 38 |
+
"is_gated_act": false,
|
| 39 |
+
"layer_norm_epsilon": 1e-06,
|
| 40 |
+
"model_type": "t5",
|
| 41 |
+
"num_heads": 8,
|
| 42 |
+
"num_layers": 6,
|
| 43 |
+
"pad_token_id": 0,
|
| 44 |
+
"reg_token_id": 1,
|
| 45 |
+
"rope_theta": 10000.0,
|
| 46 |
+
"torch_dtype": "float32",
|
| 47 |
+
"transformers_version": "4.49.0",
|
| 48 |
+
"use_cache": true,
|
| 49 |
+
"vocab_size": 2
|
| 50 |
+
}
|
model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:492290ae82bb89f9769e3479ce90b3179de1f33e600c34daa0352531538b23cd
|
| 3 |
+
size 111749048
|