diff --git a/Train_Colab.ipynb b/Train_Colab.ipynb
index b8a2bff..0b50fa8 100644
--- a/Train_Colab.ipynb
+++ b/Train_Colab.ipynb
@@ -102,7 +102,7 @@
         "!pip install -q wandb==0.13.6\n",
         "!pip install -q pyre-extensions==0.0.23\n",
         "if \"A100\" in s:\n",
-        "  !pip install -q https://huggingface.co/industriaditat/xformers_precompiles/blob/main/A100_13dev/xformers-0.0.13.dev0-py3-none-any.whl\n",
+        "  !pip install -q https://huggingface.co/industriaditat/xformers_precompiles/resolve/main/A100_13dev/xformers-0.0.13.dev0-py3-none-any.whl\n",
         "else:\n",
         "  !pip install -q https://huggingface.co/industriaditat/xformers_precompiles/resolve/main/T4_13dev/xformers-0.0.13.dev0-py3-none-any.whl\n",
         "!pip install -q pytorch-lightning==1.6.5\n",
@@ -110,10 +110,10 @@
         "!pip install -q numpy==1.23.5\n",
         "!pip install -q colorama\n",
         "!pip install -q keyboard\n",
+        "!pip install -q triton\n",
         "clear_output()\n",
         "!git clone https://github.com/victorchall/EveryDream2trainer.git\n",
         "%cd /content/EveryDream2trainer\n",
-        "!wget \"https://raw.githubusercontent.com/nawnie/EveryDream2trainer/main/train_colab.py\"\n",
         "clear_output()\n",
         "print(\"DONE!\")"
       ]