From fad9d7a32296c4e104e6462b978a77195d0f47ee Mon Sep 17 00:00:00 2001
From: SillyLossy <titan0599@gmail.com>
Date: Thu, 6 Apr 2023 22:28:21 +0300
Subject: [PATCH] Colab again

---
 colab/GPU.ipynb | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/colab/GPU.ipynb b/colab/GPU.ipynb
index 251c0d643..9ed510ab9 100644
--- a/colab/GPU.ipynb
+++ b/colab/GPU.ipynb
@@ -226,7 +226,10 @@
         "  \"Dolly_GPT-J-6b\": ModelData(\"TehVenom/Dolly_GPT-J-6b\"),\n",
         "  \"Dolly_Pyg-6B\": ModelData(\"TehVenom/AvgMerge_Dolly-Pygmalion-6b\")\n",
         "}\n",
-        "model = models.get(Model)\n",
+        "model = models.get(Model, None)\n",
+        "\n",
+        "if model == None:\n",
+        "    model = models.get(base64.b64encode(Model.encode(\"ascii\")).decode(\"ascii\"), ModelData(Model, Version))\n",
         "\n",
         "if StartKoboldAI:\n",
         "  def downloadKobold():\n",