Instructions to use ecmwf/aifs-single-2.0 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- AnemoI
How to use ecmwf/aifs-single-2.0 with AnemoI:
from anemoi.inference.runners.default import DefaultRunner from anemoi.inference.config.run import RunConfiguration # Create Configuration config = RunConfiguration(checkpoint = {"huggingface":"ecmwf/aifs-single-2.0"}) # Load Runner runner = DefaultRunner(config) - Notebooks
- Google Colab
- Kaggle
Commit ·
08286fc
1
Parent(s): 1b8913a
fix: Update anemoi-utils to 0.4.35.post3
Browse files- pyproject.toml +1 -1
- run_AIFS_v2.0.ipynb +2 -2
- uv.lock +10 -12
pyproject.toml
CHANGED
|
@@ -12,7 +12,7 @@ dependencies = [
|
|
| 12 |
"anemoi-graphs==0.6.4",
|
| 13 |
"anemoi-models==0.9.3",
|
| 14 |
"anemoi-transform==0.1.16.post2",
|
| 15 |
-
"anemoi-utils==0.4.35",
|
| 16 |
"earthkit-data<1",
|
| 17 |
"earthkit-plots<1",
|
| 18 |
"flash-attn==2.7.4",
|
|
|
|
| 12 |
"anemoi-graphs==0.6.4",
|
| 13 |
"anemoi-models==0.9.3",
|
| 14 |
"anemoi-transform==0.1.16.post2",
|
| 15 |
+
"anemoi-utils==0.4.35.post3",
|
| 16 |
"earthkit-data<1",
|
| 17 |
"earthkit-plots<1",
|
| 18 |
"flash-attn==2.7.4",
|
run_AIFS_v2.0.ipynb
CHANGED
|
@@ -91,7 +91,7 @@
|
|
| 91 |
},
|
| 92 |
"outputs": [],
|
| 93 |
"source": [
|
| 94 |
-
"# !pip install -q anemoi-inference[huggingface]==0.8.3 anemoi-models==0.9.3\n",
|
| 95 |
"# !pip install -q torch==2.7.0 torch-geometric==2.6.1\n",
|
| 96 |
"# !pip install -q earthkit-regrid==0.5.1 ecmwf-opendata==0.3.29 'earthkit-data<1.0.0'\n",
|
| 97 |
"# !pip install -q flash-attn==2.7.4.post1"
|
|
@@ -123,7 +123,7 @@
|
|
| 123 |
"metadata": {},
|
| 124 |
"outputs": [],
|
| 125 |
"source": [
|
| 126 |
-
"# %pip install anemoi-inference[huggingface]==0.8.3 anemoi-models==0.9.3\n",
|
| 127 |
"# %pip install -q torch==2.7.0 torch-geometric==2.6.1\n",
|
| 128 |
"# %pip install -q earthkit-regrid==0.5.1 ecmwf-opendata==0.3.29 'earthkit-data<1.0.0'\n",
|
| 129 |
"# %pip install -q --force-reinstall --no-deps \"flash-attn @ https://github.com/cathalobrien/get-flash-attn/releases/download/v0.1-alpha/flash_attn-2.8.3+cu12torch2.7cxx11abiFALSE-cp312-cp312-linux_x86_64.whl\""
|
|
|
|
| 91 |
},
|
| 92 |
"outputs": [],
|
| 93 |
"source": [
|
| 94 |
+
"# !pip install -q anemoi-inference[huggingface]==0.8.3 anemoi-models==0.9.3 anemoi-utils==0.4.35.post3\n",
|
| 95 |
"# !pip install -q torch==2.7.0 torch-geometric==2.6.1\n",
|
| 96 |
"# !pip install -q earthkit-regrid==0.5.1 ecmwf-opendata==0.3.29 'earthkit-data<1.0.0'\n",
|
| 97 |
"# !pip install -q flash-attn==2.7.4.post1"
|
|
|
|
| 123 |
"metadata": {},
|
| 124 |
"outputs": [],
|
| 125 |
"source": [
|
| 126 |
+
"# %pip install anemoi-inference[huggingface]==0.8.3 anemoi-models==0.9.3 anemoi-utils==0.4.35.post3\n",
|
| 127 |
"# %pip install -q torch==2.7.0 torch-geometric==2.6.1\n",
|
| 128 |
"# %pip install -q earthkit-regrid==0.5.1 ecmwf-opendata==0.3.29 'earthkit-data<1.0.0'\n",
|
| 129 |
"# %pip install -q --force-reinstall --no-deps \"flash-attn @ https://github.com/cathalobrien/get-flash-attn/releases/download/v0.1-alpha/flash_attn-2.8.3+cu12torch2.7cxx11abiFALSE-cp312-cp312-linux_x86_64.whl\""
|
uv.lock
CHANGED
|
@@ -36,8 +36,8 @@ dependencies = [
|
|
| 36 |
{ name = "anemoi-utils", marker = "(platform_machine == 'aarch64' and sys_platform == 'linux') or (platform_machine == 'x86_64' and sys_platform == 'linux')" },
|
| 37 |
{ name = "earthkit-data", marker = "(platform_machine == 'aarch64' and sys_platform == 'linux') or (platform_machine == 'x86_64' and sys_platform == 'linux')" },
|
| 38 |
{ name = "earthkit-plots", marker = "(platform_machine == 'aarch64' and sys_platform == 'linux') or (platform_machine == 'x86_64' and sys_platform == 'linux')" },
|
| 39 |
-
{ name = "flash-attn", version = "2.8.3", source = { url = "https://github.com/cathalobrien/get-flash-attn/releases/download/v0.1-alpha/flash_attn-2.8.3+cu12torch2.7cxx11abiFALSE-cp312-cp312-linux_aarch64.whl" }, marker = "platform_machine == 'aarch64' and sys_platform == 'linux'" },
|
| 40 |
-
{ name = "flash-attn", version = "2.8.3", source = { url = "https://github.com/cathalobrien/get-flash-attn/releases/download/v0.1-alpha/flash_attn-2.8.3+cu12torch2.7cxx11abiFALSE-cp312-cp312-linux_x86_64.whl" }, marker = "platform_machine == 'x86_64' and sys_platform == 'linux'" },
|
| 41 |
{ name = "torch", marker = "(platform_machine == 'aarch64' and sys_platform == 'linux') or (platform_machine == 'x86_64' and sys_platform == 'linux')" },
|
| 42 |
{ name = "torch-geometric", marker = "(platform_machine == 'aarch64' and sys_platform == 'linux') or (platform_machine == 'x86_64' and sys_platform == 'linux')" },
|
| 43 |
]
|
|
@@ -64,7 +64,7 @@ requires-dist = [
|
|
| 64 |
{ name = "anemoi-plugins-ecmwf-inference", extras = ["opendata"], marker = "extra == 'inference'", specifier = "==0.2.1" },
|
| 65 |
{ name = "anemoi-training", marker = "extra == 'training'", specifier = "==0.6.3" },
|
| 66 |
{ name = "anemoi-transform", specifier = "==0.1.16.post2" },
|
| 67 |
-
{ name = "anemoi-utils", specifier = "==0.4.35" },
|
| 68 |
{ name = "earthkit-data", specifier = "<1" },
|
| 69 |
{ name = "earthkit-plots", specifier = "<1" },
|
| 70 |
{ name = "earthkit-regrid", marker = "extra == 'inference'", specifier = "==0.5.1" },
|
|
@@ -294,7 +294,7 @@ wheels = [
|
|
| 294 |
|
| 295 |
[[package]]
|
| 296 |
name = "anemoi-utils"
|
| 297 |
-
version = "0.4.35"
|
| 298 |
source = { registry = "https://pypi.org/simple" }
|
| 299 |
dependencies = [
|
| 300 |
{ name = "aniso8601", marker = "(platform_machine == 'aarch64' and sys_platform == 'linux') or (platform_machine == 'x86_64' and sys_platform == 'linux')" },
|
|
@@ -307,9 +307,9 @@ dependencies = [
|
|
| 307 |
{ name = "rich", marker = "(platform_machine == 'aarch64' and sys_platform == 'linux') or (platform_machine == 'x86_64' and sys_platform == 'linux')" },
|
| 308 |
{ name = "tqdm", marker = "(platform_machine == 'aarch64' and sys_platform == 'linux') or (platform_machine == 'x86_64' and sys_platform == 'linux')" },
|
| 309 |
]
|
| 310 |
-
sdist = { url = "https://files.pythonhosted.org/packages/
|
| 311 |
wheels = [
|
| 312 |
-
{ url = "https://files.pythonhosted.org/packages/
|
| 313 |
]
|
| 314 |
|
| 315 |
[package.optional-dependencies]
|
|
@@ -1108,11 +1108,10 @@ version = "2.8.3"
|
|
| 1108 |
source = { url = "https://github.com/cathalobrien/get-flash-attn/releases/download/v0.1-alpha/flash_attn-2.8.3+cu12torch2.7cxx11abiFALSE-cp312-cp312-linux_aarch64.whl" }
|
| 1109 |
resolution-markers = [
|
| 1110 |
"python_full_version >= '3.12' and platform_machine == 'aarch64' and sys_platform == 'linux'",
|
| 1111 |
-
"python_full_version < '3.12' and platform_machine == 'aarch64' and sys_platform == 'linux'",
|
| 1112 |
]
|
| 1113 |
dependencies = [
|
| 1114 |
-
{ name = "einops", marker = "platform_machine == 'aarch64' and sys_platform == 'linux'" },
|
| 1115 |
-
{ name = "torch", marker = "platform_machine == 'aarch64' and sys_platform == 'linux'" },
|
| 1116 |
]
|
| 1117 |
wheels = [
|
| 1118 |
{ url = "https://github.com/cathalobrien/get-flash-attn/releases/download/v0.1-alpha/flash_attn-2.8.3+cu12torch2.7cxx11abiFALSE-cp312-cp312-linux_aarch64.whl", hash = "sha256:9dabca7ac35b5b62773318c0a2124605841df4ceba107a9a508691699b999bcc" },
|
|
@@ -1130,11 +1129,10 @@ version = "2.8.3"
|
|
| 1130 |
source = { url = "https://github.com/cathalobrien/get-flash-attn/releases/download/v0.1-alpha/flash_attn-2.8.3+cu12torch2.7cxx11abiFALSE-cp312-cp312-linux_x86_64.whl" }
|
| 1131 |
resolution-markers = [
|
| 1132 |
"python_full_version >= '3.12' and platform_machine == 'x86_64' and sys_platform == 'linux'",
|
| 1133 |
-
"python_full_version < '3.12' and platform_machine == 'x86_64' and sys_platform == 'linux'",
|
| 1134 |
]
|
| 1135 |
dependencies = [
|
| 1136 |
-
{ name = "einops", marker = "platform_machine == 'x86_64' and sys_platform == 'linux'" },
|
| 1137 |
-
{ name = "torch", marker = "platform_machine == 'x86_64' and sys_platform == 'linux'" },
|
| 1138 |
]
|
| 1139 |
wheels = [
|
| 1140 |
{ url = "https://github.com/cathalobrien/get-flash-attn/releases/download/v0.1-alpha/flash_attn-2.8.3+cu12torch2.7cxx11abiFALSE-cp312-cp312-linux_x86_64.whl", hash = "sha256:21c74313440453b914aa282a7bc78c8944dbe3b6c44989c9b36b64cdd345c933" },
|
|
|
|
| 36 |
{ name = "anemoi-utils", marker = "(platform_machine == 'aarch64' and sys_platform == 'linux') or (platform_machine == 'x86_64' and sys_platform == 'linux')" },
|
| 37 |
{ name = "earthkit-data", marker = "(platform_machine == 'aarch64' and sys_platform == 'linux') or (platform_machine == 'x86_64' and sys_platform == 'linux')" },
|
| 38 |
{ name = "earthkit-plots", marker = "(platform_machine == 'aarch64' and sys_platform == 'linux') or (platform_machine == 'x86_64' and sys_platform == 'linux')" },
|
| 39 |
+
{ name = "flash-attn", version = "2.8.3", source = { url = "https://github.com/cathalobrien/get-flash-attn/releases/download/v0.1-alpha/flash_attn-2.8.3+cu12torch2.7cxx11abiFALSE-cp312-cp312-linux_aarch64.whl" }, marker = "python_full_version >= '3.12' and platform_machine == 'aarch64' and sys_platform == 'linux'" },
|
| 40 |
+
{ name = "flash-attn", version = "2.8.3", source = { url = "https://github.com/cathalobrien/get-flash-attn/releases/download/v0.1-alpha/flash_attn-2.8.3+cu12torch2.7cxx11abiFALSE-cp312-cp312-linux_x86_64.whl" }, marker = "python_full_version >= '3.12' and platform_machine == 'x86_64' and sys_platform == 'linux'" },
|
| 41 |
{ name = "torch", marker = "(platform_machine == 'aarch64' and sys_platform == 'linux') or (platform_machine == 'x86_64' and sys_platform == 'linux')" },
|
| 42 |
{ name = "torch-geometric", marker = "(platform_machine == 'aarch64' and sys_platform == 'linux') or (platform_machine == 'x86_64' and sys_platform == 'linux')" },
|
| 43 |
]
|
|
|
|
| 64 |
{ name = "anemoi-plugins-ecmwf-inference", extras = ["opendata"], marker = "extra == 'inference'", specifier = "==0.2.1" },
|
| 65 |
{ name = "anemoi-training", marker = "extra == 'training'", specifier = "==0.6.3" },
|
| 66 |
{ name = "anemoi-transform", specifier = "==0.1.16.post2" },
|
| 67 |
+
{ name = "anemoi-utils", specifier = "==0.4.35.post3" },
|
| 68 |
{ name = "earthkit-data", specifier = "<1" },
|
| 69 |
{ name = "earthkit-plots", specifier = "<1" },
|
| 70 |
{ name = "earthkit-regrid", marker = "extra == 'inference'", specifier = "==0.5.1" },
|
|
|
|
| 294 |
|
| 295 |
[[package]]
|
| 296 |
name = "anemoi-utils"
|
| 297 |
+
version = "0.4.35.post3"
|
| 298 |
source = { registry = "https://pypi.org/simple" }
|
| 299 |
dependencies = [
|
| 300 |
{ name = "aniso8601", marker = "(platform_machine == 'aarch64' and sys_platform == 'linux') or (platform_machine == 'x86_64' and sys_platform == 'linux')" },
|
|
|
|
| 307 |
{ name = "rich", marker = "(platform_machine == 'aarch64' and sys_platform == 'linux') or (platform_machine == 'x86_64' and sys_platform == 'linux')" },
|
| 308 |
{ name = "tqdm", marker = "(platform_machine == 'aarch64' and sys_platform == 'linux') or (platform_machine == 'x86_64' and sys_platform == 'linux')" },
|
| 309 |
]
|
| 310 |
+
sdist = { url = "https://files.pythonhosted.org/packages/be/4d/c31319293fb0a6fb0160d59e8e4b0a4233c5c5db95c8dedf6fe76582a505/anemoi_utils-0.4.35.post3.tar.gz", hash = "sha256:4ec1bc9e3843f98dd678d41db79572c3a73cc4752fb9bc76a5f74e3ed8ff494c", size = 142205, upload-time = "2026-05-18T14:10:53.654Z" }
|
| 311 |
wheels = [
|
| 312 |
+
{ url = "https://files.pythonhosted.org/packages/03/37/5b593014575e89354bf4e35d921b2f569b3c35f8ba5d8ec7142eb1ce2383/anemoi_utils-0.4.35.post3-py3-none-any.whl", hash = "sha256:8be900064d2d081ac5a712905f4cca3ec3895e5d3c2dde309d153e50e52afbe6", size = 92252, upload-time = "2026-05-18T14:10:52.311Z" },
|
| 313 |
]
|
| 314 |
|
| 315 |
[package.optional-dependencies]
|
|
|
|
| 1108 |
source = { url = "https://github.com/cathalobrien/get-flash-attn/releases/download/v0.1-alpha/flash_attn-2.8.3+cu12torch2.7cxx11abiFALSE-cp312-cp312-linux_aarch64.whl" }
|
| 1109 |
resolution-markers = [
|
| 1110 |
"python_full_version >= '3.12' and platform_machine == 'aarch64' and sys_platform == 'linux'",
|
|
|
|
| 1111 |
]
|
| 1112 |
dependencies = [
|
| 1113 |
+
{ name = "einops", marker = "python_full_version >= '3.12' and platform_machine == 'aarch64' and sys_platform == 'linux'" },
|
| 1114 |
+
{ name = "torch", marker = "python_full_version >= '3.12' and platform_machine == 'aarch64' and sys_platform == 'linux'" },
|
| 1115 |
]
|
| 1116 |
wheels = [
|
| 1117 |
{ url = "https://github.com/cathalobrien/get-flash-attn/releases/download/v0.1-alpha/flash_attn-2.8.3+cu12torch2.7cxx11abiFALSE-cp312-cp312-linux_aarch64.whl", hash = "sha256:9dabca7ac35b5b62773318c0a2124605841df4ceba107a9a508691699b999bcc" },
|
|
|
|
| 1129 |
source = { url = "https://github.com/cathalobrien/get-flash-attn/releases/download/v0.1-alpha/flash_attn-2.8.3+cu12torch2.7cxx11abiFALSE-cp312-cp312-linux_x86_64.whl" }
|
| 1130 |
resolution-markers = [
|
| 1131 |
"python_full_version >= '3.12' and platform_machine == 'x86_64' and sys_platform == 'linux'",
|
|
|
|
| 1132 |
]
|
| 1133 |
dependencies = [
|
| 1134 |
+
{ name = "einops", marker = "python_full_version >= '3.12' and platform_machine == 'x86_64' and sys_platform == 'linux'" },
|
| 1135 |
+
{ name = "torch", marker = "python_full_version >= '3.12' and platform_machine == 'x86_64' and sys_platform == 'linux'" },
|
| 1136 |
]
|
| 1137 |
wheels = [
|
| 1138 |
{ url = "https://github.com/cathalobrien/get-flash-attn/releases/download/v0.1-alpha/flash_attn-2.8.3+cu12torch2.7cxx11abiFALSE-cp312-cp312-linux_x86_64.whl", hash = "sha256:21c74313440453b914aa282a7bc78c8944dbe3b6c44989c9b36b64cdd345c933" },
|