diff --git a/Train_Colab.ipynb b/Train_Colab.ipynb
index 4b93dfd..1bb7a67 100644
--- a/Train_Colab.ipynb
+++ b/Train_Colab.ipynb
@@ -3,8 +3,8 @@
     {
       "cell_type": "markdown",
       "metadata": {
-        "id": "view-in-github",
-        "colab_type": "text"
+        "colab_type": "text",
+        "id": "view-in-github"
       },
       "source": [
         "<a href=\"https://colab.research.google.com/github/nawnie/EveryDream2trainer/blob/main/Train_Colab.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
@@ -151,7 +151,8 @@
         "    'colorama',\n",
         "    'keyboard',\n",
         "    'triton',\n",
-        "    'lion-pytorch'\n",
+        "    'lion-pytorch',\n",
+        "    'compel'\n",
         "]\n",
         "\n",
         "for package in tqdm(packages, desc='Installing packages', unit='package'):\n",
@@ -269,8 +270,8 @@
       "cell_type": "code",
       "execution_count": null,
       "metadata": {
-        "id": "j9pEI69WXS9w",
-        "cellView": "form"
+        "cellView": "form",
+        "id": "j9pEI69WXS9w"
       },
       "outputs": [],
       "source": [
@@ -566,8 +567,8 @@
   "metadata": {
     "accelerator": "GPU",
     "colab": {
-      "provenance": [],
-      "include_colab_link": true
+      "include_colab_link": true,
+      "provenance": []
     },
     "gpuClass": "standard",
     "kernelspec": {
@@ -587,4 +588,4 @@
   },
   "nbformat": 4,
   "nbformat_minor": 0
-}
\ No newline at end of file
+}