Skip to content

Commit ffa30ae

Browse files
committed
Update quick start
1 parent c6af5e2 commit ffa30ae

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

notebooks/chronos-2-quickstart.ipynb

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,12 @@
77
"source": [
88
"# Getting Started with Chronos-2\n",
99
"\n",
10+
"[![Open In SageMaker Studio Lab](https://studiolab.sagemaker.aws/studiolab.svg)](https://studiolab.sagemaker.aws/import/github/amazon-science/chronos-forecasting/blob/main/notebooks/chronos-2-quickstart.ipynb)\n",
11+
"[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](\n",
12+
"https://colab.research.google.com/github/amazon-science/chronos-forecasting/blob/main/notebooks/chronos-2-quickstart.ipynb)\n",
13+
"\n",
14+
"\n",
15+
"\n",
1016
"**Chronos-2** is a foundation model for time series forecasting that builds on [Chronos](https://arxiv.org/abs/2403.07815) and [Chronos-Bolt](https://aws.amazon.com/blogs/machine-learning/fast-and-accurate-zero-shot-forecasting-with-chronos-bolt-and-autogluon/). It offers significant improvements in capabilities and can handle diverse forecasting scenarios not supported by earlier models.\n",
1117
"\n",
1218
"| Capability | Chronos | Chronos-Bolt | Chronos-2 |\n",
@@ -31,7 +37,7 @@
3137
"metadata": {},
3238
"outputs": [],
3339
"source": [
34-
"%pip install -U 'chronos-forecasting>=2.0' 'pandas[pyarrow]' 'matplotlib'"
40+
"%pip install 'chronos-forecasting>=2.0' 'pandas[pyarrow]' 'matplotlib'"
3541
]
3642
},
3743
{
@@ -41,9 +47,9 @@
4147
"metadata": {},
4248
"outputs": [],
4349
"source": [
44-
"# Use only 1 GPU if available\n",
4550
"import os\n",
4651
"\n",
52+
"# Use only 1 GPU if available\n",
4753
"os.environ[\"CUDA_VISIBLE_DEVICES\"] = \"0\"\n",
4854
"\n",
4955
"import pandas as pd\n",
@@ -1586,7 +1592,7 @@
15861592
"name": "python",
15871593
"nbconvert_exporter": "python",
15881594
"pygments_lexer": "ipython3",
1589-
"version": "3.11.11"
1595+
"version": "3.11.13"
15901596
}
15911597
},
15921598
"nbformat": 4,

0 commit comments

Comments
 (0)