diff --git a/Train_Colab.ipynb b/Train_Colab.ipynb
index 495035b..bbe89e6 100644
--- a/Train_Colab.ipynb
+++ b/Train_Colab.ipynb
@@ -82,8 +82,8 @@
       "cell_type": "code",
       "execution_count": null,
       "metadata": {
-        "id": "hAuBbtSvGpau",
-        "cellView": "form"
+        "cellView": "form",
+        "id": "hAuBbtSvGpau"
       },
       "outputs": [],
       "source": [
@@ -110,6 +110,7 @@
         "!pip install -q colorama\n",
         "!pip install -q keyboard\n",
         "!pip install -q triton\n",
+        "!pip install -q lion-pytorch\n",
         "clear_output()\n",
         "!git clone https://github.com/victorchall/EveryDream2trainer.git\n",
         "%cd /content/EveryDream2trainer\n",
@@ -506,4 +507,4 @@
   },
   "nbformat": 4,
   "nbformat_minor": 0
-}
\ No newline at end of file
+}