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",