Update README.md
Browse files
README.md
CHANGED
@@ -55,7 +55,7 @@ and direct observational data.
|
|
55 |
## How to Get Started with the Model
|
56 |
|
57 |
To be able to run AIFS to generate a new forecast, you can use ai-models https://github.com/ecmwf-lab/ai-models.
|
58 |
-
`ai-models` command can be used to run different models, since in this case we are looking at using AIFS we need to
|
59 |
`anemoi` as `model-name` and then pass the path to the checkpoint (`aifs_single_v0.2.1.ckpt`) and the initial conditions.
|
60 |
You can find an example of a set of initial conditions in the GRIB file `example_20241107_12_n320.grib`.
|
61 |
|
@@ -63,7 +63,7 @@ Use the code below to get started with the model.
|
|
63 |
|
64 |
|
65 |
```
|
66 |
-
# 1st create the conda environment
|
67 |
export CONDA_ENV=aifs-env
|
68 |
conda create -n ${CONDA_ENV} python=3.10
|
69 |
conda activate ${CONDA_ENV}
|
@@ -73,14 +73,14 @@ pip install anemoi-inference[plugin] anemoi-models==0.2
|
|
73 |
pip install ninja
|
74 |
pip install flash-attn --no-build-isolation
|
75 |
|
76 |
-
# 2nd Run ai-models to generate weather forecast
|
77 |
|
78 |
ai-models anemoi --checkpoint aifs_single_v0.2.1.ckpt --file example_20241107_12_n320.grib
|
79 |
```
|
80 |
|
81 |
-
**Note** we train AIFS using `flash_attention` (https://github.com/Dao-AILab/flash-attention)
|
82 |
-
There are currently some issues when trying to install flash attention with the latest PyTorch version 2.5 and CUDA 12.4 (https://github.com/Dao-AILab/flash-attention/issues/1330)
|
83 |
-
For that reason we
|
84 |
|
85 |
After running the `ai-models` command the output of the forecast should be written into `anemoi.grib`
|
86 |
Below you can find an example to read that file and load it as numpy array or xarray.
|
|
|
55 |
## How to Get Started with the Model
|
56 |
|
57 |
To be able to run AIFS to generate a new forecast, you can use ai-models https://github.com/ecmwf-lab/ai-models.
|
58 |
+
`ai-models` command can be used to run different models, since in this case we are looking at using AIFS we need to specify
|
59 |
`anemoi` as `model-name` and then pass the path to the checkpoint (`aifs_single_v0.2.1.ckpt`) and the initial conditions.
|
60 |
You can find an example of a set of initial conditions in the GRIB file `example_20241107_12_n320.grib`.
|
61 |
|
|
|
63 |
|
64 |
|
65 |
```
|
66 |
+
# 1st - create the conda environment
|
67 |
export CONDA_ENV=aifs-env
|
68 |
conda create -n ${CONDA_ENV} python=3.10
|
69 |
conda activate ${CONDA_ENV}
|
|
|
73 |
pip install ninja
|
74 |
pip install flash-attn --no-build-isolation
|
75 |
|
76 |
+
# 2nd - Run ai-models to generate weather forecast
|
77 |
|
78 |
ai-models anemoi --checkpoint aifs_single_v0.2.1.ckpt --file example_20241107_12_n320.grib
|
79 |
```
|
80 |
|
81 |
+
**Note** we train AIFS using `flash_attention` (https://github.com/Dao-AILab/flash-attention).
|
82 |
+
There are currently some issues when trying to install flash attention with the latest PyTorch version 2.5 and CUDA 12.4 (https://github.com/Dao-AILab/flash-attention/issues/1330).
|
83 |
+
For that reason, we recommend you install PyTorch 2.4.
|
84 |
|
85 |
After running the `ai-models` command the output of the forecast should be written into `anemoi.grib`
|
86 |
Below you can find an example to read that file and load it as numpy array or xarray.
|