File size: 1,448 Bytes
38fd300 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 | [project]
name = "aifs-ens"
version = "2.0"
requires-python = ">=3.11,<3.14"
classifiers = [
"Programming Language :: Python :: 3 :: Only",
"Programming Language :: Python :: 3.11",
"Programming Language :: Python :: 3.12",
"Programming Language :: Python :: 3.13",
]
dependencies = [
"anemoi-graphs==0.8.1",
"anemoi-models==0.11.2",
"anemoi-transform==0.1.16.post2",
"anemoi-utils==0.4.37",
"earthkit-data<1",
"earthkit-plots<1",
"flash-attn==2.7.4",
"torch==2.7",
"torch-geometric==2.6.1",
]
optional-dependencies.inference = [
"anemoi-inference[huggingface]==0.8.3",
"anemoi-plugins-ecmwf-inference[opendata]==0.2.1",
"earthkit-regrid==0.5.1",
"ecmwf-opendata==0.3.29",
]
optional-dependencies.jupyter = [
"ipykernel>=7.2",
]
optional-dependencies.training = [
"anemoi-training==0.8.1",
]
[tool.uv]
sources.flash-attn = [
{ 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 != 'win32'" },
{ 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'" },
]
environments = [
"sys_platform == 'linux' and (platform_machine == 'x86_64' or platform_machine == 'aarch64')",
]
|