From 072cfe19e9ea9bcf583c265333975304e85baec6 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Wed, 15 Nov 2023 08:18:32 -0800
Subject: [PATCH] Minor Colab fix

---
 Colab-TextGen-GPU.ipynb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Colab-TextGen-GPU.ipynb b/Colab-TextGen-GPU.ipynb
index e1950b75..f8b6719f 100644
--- a/Colab-TextGen-GPU.ipynb
+++ b/Colab-TextGen-GPU.ipynb
@@ -106,7 +106,7 @@
         "    url_parts = model_url.strip('/').strip().split('/')\n",
         "    output_folder = f\"{url_parts[-2]}_{url_parts[-1]}\"\n",
         "    branch = branch.strip('\"\\' ')\n",
-        "    if branch.strip() != not in ['', 'main']:\n",
+        "    if branch.strip() not in ['', 'main']:\n",
         "        output_folder += f\"_{branch}\"\n",
         "        !python download-model.py {model_url} --branch {branch}\n",
         "    else:\n",