Spaces:
Build error
Build error
add flash attn
Browse files- requirements.txt +6 -2
requirements.txt
CHANGED
|
@@ -1,9 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
| 1 |
anemoi-inference[huggingface]==0.4.9
|
| 2 |
anemoi-models==0.3.1
|
| 3 |
earthkit-regrid==0.4.0
|
| 4 |
ecmwf-opendata
|
| 5 |
-
flash-attn
|
| 6 |
gradio
|
| 7 |
numpy
|
| 8 |
matplotlib
|
| 9 |
-
cartopy
|
|
|
|
|
|
|
|
|
| 1 |
+
--extra-index-url https://download.pytorch.org/whl/cu118
|
| 2 |
+
torch>=2.0.0
|
| 3 |
+
flash-attn==2.7.4.post1
|
| 4 |
anemoi-inference[huggingface]==0.4.9
|
| 5 |
anemoi-models==0.3.1
|
| 6 |
earthkit-regrid==0.4.0
|
| 7 |
ecmwf-opendata
|
|
|
|
| 8 |
gradio
|
| 9 |
numpy
|
| 10 |
matplotlib
|
| 11 |
+
cartopy
|
| 12 |
+
xarray
|
| 13 |
+
netCDF4
|