ecmwf-aifs-space / requirements.txt
saburq's picture
add flash attn
75d82f6
raw
history blame
236 Bytes
--extra-index-url https://download.pytorch.org/whl/cu118
torch>=2.0.0
flash-attn==2.7.4.post1
anemoi-inference[huggingface]==0.4.9
anemoi-models==0.3.1
earthkit-regrid==0.4.0
ecmwf-opendata
gradio
numpy
matplotlib
cartopy
xarray
netCDF4