diff --git a/Train_Colab.ipynb b/Train_Colab.ipynb
index fc0f32e..ced6b81 100644
--- a/Train_Colab.ipynb
+++ b/Train_Colab.ipynb
@@ -82,7 +82,6 @@
       "cell_type": "code",
       "execution_count": null,
       "metadata": {
-        "cellView": "form",
         "id": "hAuBbtSvGpau"
       },
       "outputs": [],
@@ -97,6 +96,7 @@
         "!pip install -q diffusers[torch]==0.13.0\n",
         "!pip install -q pynvml==11.4.1\n",
         "!pip install -q bitsandbytes==0.35.0\n",
+        "!pip install -q lion \n",
         "!pip install -q ftfy==6.1.1\n",
         "!pip install -q aiohttp==3.8.3\n",
         "!pip install -q tensorboard>=2.11.0\n",