Graph Machine Learning
AnemoI
English
hcookie129 commited on
Commit
f0bb02c
·
1 Parent(s): e417b59

fix: Add torch to install cell

Browse files
Files changed (1) hide show
  1. run_AIFS_v1.ipynb +4 -4
run_AIFS_v1.ipynb CHANGED
@@ -20,7 +20,7 @@
20
  },
21
  {
22
  "cell_type": "code",
23
- "execution_count": 1,
24
  "metadata": {
25
  "colab": {
26
  "base_uri": "https://localhost:8080/"
@@ -42,7 +42,7 @@
42
  "source": [
43
  "# Uncomment the lines below to install the required packages\n",
44
  "\n",
45
- "# !pip install -q anemoi-inference[huggingface]==0.4.9 anemoi-models==0.3.1\n",
46
  "# !pip install -q earthkit-regrid==0.4.0 ecmwf-opendata \n",
47
  "# !pip install -q flash_attn"
48
  ]
@@ -791,6 +791,7 @@
791
  "pygments_lexer": "ipython3",
792
  "version": "3.11.10"
793
  },
 
794
  "widgets": {
795
  "application/vnd.jupyter.widget-state+json": {
796
  "00131dfdac6c4787b3c6aefb08f13beb": {
@@ -2846,8 +2847,7 @@
2846
  }
2847
  }
2848
  }
2849
- },
2850
- "license": "Apache-2.0"
2851
  },
2852
  "nbformat": 4,
2853
  "nbformat_minor": 4
 
20
  },
21
  {
22
  "cell_type": "code",
23
+ "execution_count": null,
24
  "metadata": {
25
  "colab": {
26
  "base_uri": "https://localhost:8080/"
 
42
  "source": [
43
  "# Uncomment the lines below to install the required packages\n",
44
  "\n",
45
+ "# !pip install -q anemoi-inference[huggingface]==0.4.9 anemoi-models==0.3.1 torch==2.4.0\n",
46
  "# !pip install -q earthkit-regrid==0.4.0 ecmwf-opendata \n",
47
  "# !pip install -q flash_attn"
48
  ]
 
791
  "pygments_lexer": "ipython3",
792
  "version": "3.11.10"
793
  },
794
+ "license": "Apache-2.0",
795
  "widgets": {
796
  "application/vnd.jupyter.widget-state+json": {
797
  "00131dfdac6c4787b3c6aefb08f13beb": {
 
2847
  }
2848
  }
2849
  }
2850
+ }
 
2851
  },
2852
  "nbformat": 4,
2853
  "nbformat_minor": 4