Upload folder using huggingface_hub
Browse files- README.md +51 -1
- chronos-2.metadata.json +42 -0
- chronos-2.onnx +3 -0
- chronos-2_int8.metadata.json +42 -0
- chronos-2_int8.onnx +3 -0
- chronos-2_static.metadata.json +42 -0
- chronos-2_static.onnx +3 -0
README.md
CHANGED
|
@@ -1,3 +1,53 @@
|
|
| 1 |
---
|
| 2 |
-
license:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
+
license: apache-2.0
|
| 3 |
+
library_name: onnx
|
| 4 |
+
tags:
|
| 5 |
+
- time-series-forecasting
|
| 6 |
+
- chronos
|
| 7 |
+
- onnx
|
| 8 |
+
- openstef
|
| 9 |
---
|
| 10 |
+
|
| 11 |
+
# chronos-2 — ONNX checkpoints
|
| 12 |
+
|
| 13 |
+
ONNX export of [`amazon/chronos-2`](https://huggingface.co/amazon/chronos-2) for
|
| 14 |
+
[OpenSTEF](https://github.com/OpenSTEF/openstef), produced by
|
| 15 |
+
[`openstef-checkpoints`](https://github.com/OpenSTEF/openstef-checkpoints).
|
| 16 |
+
|
| 17 |
+
OpenSTEF resolves these via `HubCheckpoint`; the metadata file next to each weights file drives the
|
| 18 |
+
inference path. Each variant was checked against the torch reference on representative inputs before
|
| 19 |
+
publishing.
|
| 20 |
+
|
| 21 |
+
## Variants
|
| 22 |
+
|
| 23 |
+
| File | Precision | Static shapes | Max deviation vs torch |
|
| 24 |
+
|---|---|---|---|
|
| 25 |
+
| `chronos-2_static.onnx` | fp32 | yes | 1.788e-07 |
|
| 26 |
+
| `chronos-2.onnx` | fp32 | no | 2.98e-08 |
|
| 27 |
+
| `chronos-2_int8.onnx` | int8 | no | 9.677e-05 |
|
| 28 |
+
|
| 29 |
+
|
| 30 |
+
Pick by deployment target: **static fp32** is the portable, CoreML-eligible default; **int8** for size;
|
| 31 |
+
**dynamic** when context/horizon must vary.
|
| 32 |
+
|
| 33 |
+
## Provenance
|
| 34 |
+
|
| 35 |
+
- Source model: `amazon/chronos-2` @ `unknown`
|
| 36 |
+
- Exporter: `openstef-checkpoints` @ `f1ea4ab27067a7ce8b8ec79d6dd0ae9656428e11`
|
| 37 |
+
- Tooling: onnx=1.22.0 onnxruntime=1.27.0 torch=2.12.1+cpu
|
| 38 |
+
- Exported: 2026-06-19T11:18:21+00:00
|
| 39 |
+
|
| 40 |
+
## License and attribution
|
| 41 |
+
|
| 42 |
+
These ONNX checkpoints are derived from
|
| 43 |
+
[`amazon/chronos-2`](https://huggingface.co/amazon/chronos-2) and released under the same
|
| 44 |
+
license, **apache-2.0**. Attribution and all rights to the model weights remain with the
|
| 45 |
+
upstream authors; if you use the model in research, please cite their work.
|
| 46 |
+
|
| 47 |
+
**Modifications.** The original weights were converted from PyTorch to ONNX and exported in several
|
| 48 |
+
shape and precision variants (static or dynamic shapes; fp32, fp16, or int8). The weights were not
|
| 49 |
+
retrained, fine-tuned, or otherwise altered.
|
| 50 |
+
|
| 51 |
+
The tooling that produced these files
|
| 52 |
+
([`openstef-checkpoints`](https://github.com/OpenSTEF/openstef-checkpoints)) is licensed MPL-2.0,
|
| 53 |
+
which does not extend to the weights.
|
chronos-2.metadata.json
ADDED
|
@@ -0,0 +1,42 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"schema_version": 2,
|
| 3 |
+
"model_family": "chronos2",
|
| 4 |
+
"input_names": [
|
| 5 |
+
"context",
|
| 6 |
+
"group_ids",
|
| 7 |
+
"attention_mask",
|
| 8 |
+
"future_covariates",
|
| 9 |
+
"future_covariates_mask"
|
| 10 |
+
],
|
| 11 |
+
"output_name": "quantile_preds",
|
| 12 |
+
"native_quantiles": [
|
| 13 |
+
0.01,
|
| 14 |
+
0.05,
|
| 15 |
+
0.1,
|
| 16 |
+
0.15,
|
| 17 |
+
0.2,
|
| 18 |
+
0.25,
|
| 19 |
+
0.3,
|
| 20 |
+
0.35,
|
| 21 |
+
0.4,
|
| 22 |
+
0.45,
|
| 23 |
+
0.5,
|
| 24 |
+
0.55,
|
| 25 |
+
0.6,
|
| 26 |
+
0.65,
|
| 27 |
+
0.7,
|
| 28 |
+
0.75,
|
| 29 |
+
0.8,
|
| 30 |
+
0.85,
|
| 31 |
+
0.9,
|
| 32 |
+
0.95,
|
| 33 |
+
0.99
|
| 34 |
+
],
|
| 35 |
+
"context_length": 5760,
|
| 36 |
+
"output_patch_size": 16,
|
| 37 |
+
"horizon_patches": 42,
|
| 38 |
+
"resolution_minutes": 15,
|
| 39 |
+
"precision": "fp32",
|
| 40 |
+
"static_shapes": false,
|
| 41 |
+
"max_covariates": null
|
| 42 |
+
}
|
chronos-2.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ae6b9f66cf09e1c1bc673c6544b17633c3b7d302767b482695174a9b082ee58b
|
| 3 |
+
size 478418815
|
chronos-2_int8.metadata.json
ADDED
|
@@ -0,0 +1,42 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"schema_version": 2,
|
| 3 |
+
"model_family": "chronos2",
|
| 4 |
+
"input_names": [
|
| 5 |
+
"context",
|
| 6 |
+
"group_ids",
|
| 7 |
+
"attention_mask",
|
| 8 |
+
"future_covariates",
|
| 9 |
+
"future_covariates_mask"
|
| 10 |
+
],
|
| 11 |
+
"output_name": "quantile_preds",
|
| 12 |
+
"native_quantiles": [
|
| 13 |
+
0.01,
|
| 14 |
+
0.05,
|
| 15 |
+
0.1,
|
| 16 |
+
0.15,
|
| 17 |
+
0.2,
|
| 18 |
+
0.25,
|
| 19 |
+
0.3,
|
| 20 |
+
0.35,
|
| 21 |
+
0.4,
|
| 22 |
+
0.45,
|
| 23 |
+
0.5,
|
| 24 |
+
0.55,
|
| 25 |
+
0.6,
|
| 26 |
+
0.65,
|
| 27 |
+
0.7,
|
| 28 |
+
0.75,
|
| 29 |
+
0.8,
|
| 30 |
+
0.85,
|
| 31 |
+
0.9,
|
| 32 |
+
0.95,
|
| 33 |
+
0.99
|
| 34 |
+
],
|
| 35 |
+
"context_length": 5760,
|
| 36 |
+
"output_patch_size": 16,
|
| 37 |
+
"horizon_patches": 42,
|
| 38 |
+
"resolution_minutes": 15,
|
| 39 |
+
"precision": "int8",
|
| 40 |
+
"static_shapes": false,
|
| 41 |
+
"max_covariates": null
|
| 42 |
+
}
|
chronos-2_int8.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:80df1ae3d2fa580f6d7533cd022c71bcb938ad1aecd082bb9d7fcafd37778cb8
|
| 3 |
+
size 130694288
|
chronos-2_static.metadata.json
ADDED
|
@@ -0,0 +1,42 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"schema_version": 2,
|
| 3 |
+
"model_family": "chronos2",
|
| 4 |
+
"input_names": [
|
| 5 |
+
"context",
|
| 6 |
+
"group_ids",
|
| 7 |
+
"attention_mask",
|
| 8 |
+
"future_covariates",
|
| 9 |
+
"future_covariates_mask"
|
| 10 |
+
],
|
| 11 |
+
"output_name": "quantile_preds",
|
| 12 |
+
"native_quantiles": [
|
| 13 |
+
0.01,
|
| 14 |
+
0.05,
|
| 15 |
+
0.1,
|
| 16 |
+
0.15,
|
| 17 |
+
0.2,
|
| 18 |
+
0.25,
|
| 19 |
+
0.3,
|
| 20 |
+
0.35,
|
| 21 |
+
0.4,
|
| 22 |
+
0.45,
|
| 23 |
+
0.5,
|
| 24 |
+
0.55,
|
| 25 |
+
0.6,
|
| 26 |
+
0.65,
|
| 27 |
+
0.7,
|
| 28 |
+
0.75,
|
| 29 |
+
0.8,
|
| 30 |
+
0.85,
|
| 31 |
+
0.9,
|
| 32 |
+
0.95,
|
| 33 |
+
0.99
|
| 34 |
+
],
|
| 35 |
+
"context_length": 5760,
|
| 36 |
+
"output_patch_size": 16,
|
| 37 |
+
"horizon_patches": 42,
|
| 38 |
+
"resolution_minutes": 15,
|
| 39 |
+
"precision": "fp32",
|
| 40 |
+
"static_shapes": true,
|
| 41 |
+
"max_covariates": 3
|
| 42 |
+
}
|
chronos-2_static.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1b69f862056b97784bb4bd8a3ed8ae9e3129cd04ff9156f9f58b66c6e9fdf7f0
|
| 3 |
+
size 478170946
|