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.
|