From c936d9db4829c39735d9d69ddc31470684d5811d Mon Sep 17 00:00:00 2001
From: nawnie <106923464+nawnie@users.noreply.github.com>
Date: Wed, 1 Feb 2023 00:42:11 -0600
Subject: [PATCH] Update Train_Colab.ipynb

xformers==0.0.17.dev435" is just the version i test can do just pip install xformers, but i did not test 0.0.16 yet, this mixed with the other pr's really speeds up colab
---
 Train_Colab.ipynb | 7 ++-----
 1 file changed, 2 insertions(+), 5 deletions(-)

diff --git a/Train_Colab.ipynb b/Train_Colab.ipynb
index 5158c25..9cb6cc5 100644
--- a/Train_Colab.ipynb
+++ b/Train_Colab.ipynb
@@ -90,7 +90,7 @@
         "from IPython.display import clear_output\n",
         "from subprocess import getoutput\n",
         "s = getoutput('nvidia-smi')\n",
-        "!pip install -q torch==1.12.1+cu116 torchvision==0.13.1+cu116 --extra-index-url \"https://download.pytorch.org/whl/cu116\"\n",
+        "!pip install -q torch==1.13.1+cu117 torchvision==0.14.1+cu117 --extra-index-url \"https://download.pytorch.org/whl/cu117\"\n",
         "!pip install -q transformers==4.25.1\n",
         "!pip install -q diffusers[torch]==0.10.2\n",
         "!pip install -q pynvml==11.4.1\n",
@@ -101,10 +101,7 @@
         "!pip install -q protobuf==3.20.1\n",
         "!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/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 xformers==0.0.17.dev435",
         "!pip install -q pytorch-lightning==1.6.5\n",
         "!pip install -q OmegaConf==2.2.3\n",
         "!pip install -q numpy==1.23.5\n",