Update README.md

README.md

@@ -57,11 +57,15 @@ export CONDA_ENV=aifs-env
 conda create -n ${CONDA_ENV} python=3.10
 conda activate ${CONDA_ENV}
 
-pip install flash-attn
+pip install torch=2.4
 pip install anemoi-inference[plugin] anemoi-models==0.2
+pip install ninja
+pip install flash-attn --no-build-isolation
 
 ai-models anemoi --checkpoint aifs_single_v0.2.1.ckpt --file example_20241107_12_n320.grib
 ```
+! ISSUE WITH PYTORCH 2.5 and flash_attention, and cuda 12.4. For now keep it to torch 2.4
+
 The above command will write the forecast results to anemoi.grib
 #missing - example how to plot/open the anemoi.grib