Graph Machine Learning
AnemoI
English
jpxkqx commited on
Commit
29a9975
1 Parent(s): a28e1af

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -6
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 speficy
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 recommen 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.
 
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.