diff --git "a/run_AIFS_v0_2_1.ipynb" "b/run_AIFS_v0_2_1.ipynb" --- "a/run_AIFS_v0_2_1.ipynb" +++ "b/run_AIFS_v0_2_1.ipynb" @@ -42,14 +42,14 @@ "source": [ "# Uncomment the lines below to install the required packages\n", "\n", - "#!pip install -q anemoi-inference==0.3.3 anemoi-models==0.2.1\n", - "#!pip install -q earthkit-regrid==0.3.4 ecmwf-opendata huggingface_hub\n", - "#!pip install -q flash_attn" + "# !pip install -q anemoi-inference[huggingface]==0.4.3 anemoi-models==0.2.1\n", + "# !pip install -q earthkit-regrid==0.3.4 ecmwf-opendata\n", + "# !pip install -q flash_attn" ] }, { "cell_type": "code", - "execution_count": 3, + "execution_count": 2, "metadata": { "executionInfo": { "elapsed": 3143, @@ -66,14 +66,16 @@ "outputs": [], "source": [ "import datetime\n", + "from collections import defaultdict\n", + "\n", + "import numpy as np\n", "import earthkit.data as ekd\n", "import earthkit.regrid as ekr\n", - "import numpy as np\n", - "from collections import defaultdict\n", + "\n", "from anemoi.inference.runners.simple import SimpleRunner\n", "from anemoi.inference.outputs.printer import print_state\n", - "from ecmwf.opendata import Client as OpendataClient\n", - "from huggingface_hub import hf_hub_download" + "\n", + "from ecmwf.opendata import Client as OpendataClient" ] }, { @@ -98,7 +100,7 @@ }, { "cell_type": "code", - "execution_count": 4, + "execution_count": 3, "metadata": { "executionInfo": { "elapsed": 195, @@ -130,7 +132,7 @@ }, { "cell_type": "code", - "execution_count": 5, + "execution_count": 4, "metadata": { "executionInfo": { "elapsed": 2109, @@ -151,7 +153,7 @@ }, { "cell_type": "code", - "execution_count": 6, + "execution_count": 5, "metadata": { "colab": { "base_uri": "https://localhost:8080/" @@ -174,7 +176,7 @@ "name": "stdout", "output_type": "stream", "text": [ - "Initial date is 2024-12-05 18:00:00\n" + "Initial date is 2024-12-20 06:00:00\n" ] } ], @@ -193,7 +195,7 @@ }, { "cell_type": "code", - "execution_count": 7, + "execution_count": 6, "metadata": { "executionInfo": { "elapsed": 215, @@ -241,7 +243,7 @@ }, { "cell_type": "code", - "execution_count": 8, + "execution_count": 7, "metadata": { "executionInfo": { "elapsed": 186, @@ -271,7 +273,7 @@ }, { "cell_type": "code", - "execution_count": 9, + "execution_count": 8, "metadata": { "colab": { "base_uri": "https://localhost:8080/", @@ -341,7 +343,7 @@ }, { "cell_type": "code", - "execution_count": 10, + "execution_count": 9, "metadata": { "colab": { "base_uri": "https://localhost:8080/", @@ -400,7 +402,7 @@ }, { "cell_type": "code", - "execution_count": 11, + "execution_count": 10, "metadata": { "executionInfo": { "elapsed": 206, @@ -433,7 +435,7 @@ }, { "cell_type": "code", - "execution_count": 12, + "execution_count": 11, "metadata": { "executionInfo": { "elapsed": 187, @@ -467,12 +469,12 @@ "id": "9ozNl0kZy7wv" }, "source": [ - "### Download the Model's Checkpoint from Hugging Face" + "### Download the Model's Checkpoint from Hugging Face & create a Runner" ] }, { "cell_type": "code", - "execution_count": 13, + "execution_count": 12, "metadata": { "colab": { "base_uri": "https://localhost:8080/", @@ -506,21 +508,12 @@ }, "outputs": [], "source": [ - "checkpoint = hf_hub_download(\"ecmwf/aifs-single\", \"aifs_single_v0.2.1.ckpt\")" - ] - }, - { - "cell_type": "markdown", - "metadata": { - "id": "q0XkPwTbx9or" - }, - "source": [ - "#### Create a Runner" + "checkpoint = {\"huggingface\":\"ecmwf/aifs-single\"}" ] }, { "cell_type": "code", - "execution_count": 14, + "execution_count": 13, "metadata": { "executionInfo": { "elapsed": 193, @@ -534,7 +527,22 @@ }, "id": "FAw6hansLEth" }, - "outputs": [], + "outputs": [ + { + "data": { + "application/vnd.jupyter.widget-view+json": { + "model_id": "62c500d276a94f14ba3f2bef80688f4a", + "version_major": 2, + "version_minor": 0 + }, + "text/plain": [ + "Fetching 10 files: 0%| | 0/10 [00:00" ] @@ -684,9 +692,9 @@ "toc_visible": true }, "kernelspec": { - "display_name": "aifs_v021_p11", + "display_name": "huggingface-aifs0.2.1", "language": "python", - "name": "aifs_v021_p11" + "name": "huggingface-aifs0.2.1" }, "language_info": { "codemirror_mode": { @@ -698,7 +706,7 @@ "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", - "version": "3.11.10" + "version": "3.12.6" }, "widgets": { "application/vnd.jupyter.widget-state+json": {