Update all Colabs in the repo to use ai-edge-torch-nightly. (#4)

advaitjain · web-flow · commit 0294b2acc71b · 2024-05-15T16:00:42.000Z
BUG=http://b/340367182
diff --git a/ai_edge_torch/generative/examples/t5/t5_conversion_colab.ipynb b/ai_edge_torch/generative/examples/t5/t5_conversion_colab.ipynb
@@ -60,7 +60,7 @@
       "outputs": [],
       "source": [
         "!pip install -r https://raw.githubusercontent.com/google-ai-edge/ai-edge-torch/main/requirements.txt\n",
-        "!pip install ai-edge-torch"
+        "!pip install ai-edge-torch-nightly"
       ]
     },
     {
diff --git a/docs/pytorch_converter/getting_started.ipynb b/docs/pytorch_converter/getting_started.ipynb
@@ -61,7 +61,7 @@
       "outputs": [],
       "source": [
         "!pip install -r https://raw.githubusercontent.com/google-ai-edge/ai-edge-torch/main/requirements.txt\n",
-        "!pip install ai-edge-torch"
+        "!pip install ai-edge-torch-nightly"
       ]
     },
     {
diff --git a/test/image_classification/colab/mobile_vit_mpt.ipynb b/test/image_classification/colab/mobile_vit_mpt.ipynb
@@ -60,7 +60,7 @@
       "outputs": [],
       "source": [
         "!pip install -r https://raw.githubusercontent.com/google-ai-edge/ai-edge-torch/main/requirements.txt\n",
-        "!pip install ai-edge-torch\n",
+        "!pip install ai-edge-torch-nightly\n",
         "!pip install transformers pillow requests matplotlib mediapipe"
       ]
     },
diff --git a/test/image_classification/colab/mobile_vit_tfl.ipynb b/test/image_classification/colab/mobile_vit_tfl.ipynb
@@ -60,7 +60,7 @@
       "outputs": [],
       "source": [
         "!pip install -r https://raw.githubusercontent.com/google-ai-edge/ai-edge-torch/main/requirements.txt\n",
-        "!pip install ai-edge-torch\n",
+        "!pip install ai-edge-torch-nightly\n",
         "!pip install transformers pillow requests matplotlib"
       ]
     },
diff --git a/test/image_segmentation/colab/isnet_mpt.ipynb b/test/image_segmentation/colab/isnet_mpt.ipynb
@@ -60,7 +60,7 @@
       "outputs": [],
       "source": [
         "!pip install -r https://raw.githubusercontent.com/google-ai-edge/ai-edge-torch/main/requirements.txt\n",
-        "!pip install ai-edge-torch\n",
+        "!pip install ai-edge-torch-nightly\n",
         "!pip install pillow requests matplotlib mediapipe"
       ]
     },
diff --git a/test/image_segmentation/colab/isnet_tfl.ipynb b/test/image_segmentation/colab/isnet_tfl.ipynb
@@ -60,7 +60,7 @@
       "outputs": [],
       "source": [
         "!pip install -r https://raw.githubusercontent.com/google-ai-edge/ai-edge-torch/main/requirements.txt\n",
-        "!pip install ai-edge-torch\n",
+        "!pip install ai-edge-torch-nightly\n",
         "!pip install pillow requests matplotlib"
       ]
     },

Original file line number	Diff line number	Diff line change
`@@ -60,7 +60,7 @@`
`60`	`60`	`"outputs": [],`
`61`	`61`	`"source": [`
`62`	`62`	`"!pip install -r https://raw.githubusercontent.com/google-ai-edge/ai-edge-torch/main/requirements.txt\n",`
`63`		`- "!pip install ai-edge-torch"`
	`63`	`+ "!pip install ai-edge-torch-nightly"`
`64`	`64`	`]`
`65`	`65`	`},`
`66`	`66`	`{`
Original file line number	Diff line number	Diff line change
`@@ -61,7 +61,7 @@`
`61`	`61`	`"outputs": [],`
`62`	`62`	`"source": [`
`63`	`63`	`"!pip install -r https://raw.githubusercontent.com/google-ai-edge/ai-edge-torch/main/requirements.txt\n",`
`64`		`- "!pip install ai-edge-torch"`
	`64`	`+ "!pip install ai-edge-torch-nightly"`
`65`	`65`	`]`
`66`	`66`	`},`
`67`	`67`	`{`