diff --git a/colab/GPU.ipynb b/colab/GPU.ipynb index d979fb764..251c0d643 100644 --- a/colab/GPU.ipynb +++ b/colab/GPU.ipynb @@ -181,7 +181,7 @@ " \"/content/drive/MyDrive\"\n", " ])\n", "\n", - "model = {\n", + "models = {\n", " \"Nerys V2 6B\": ModelData(\"KoboldAI/OPT-6B-nerys-v2\"),\n", " \"Erebus 6B\": ModelData(\"KoboldAI/OPT-6.7B-Erebus\"),\n", " \"Skein 6B\": ModelData(\"KoboldAI/GPT-J-6B-Skein\"),\n", @@ -226,7 +226,7 @@ " \"Dolly_GPT-J-6b\": ModelData(\"TehVenom/Dolly_GPT-J-6b\"),\n", " \"Dolly_Pyg-6B\": ModelData(\"TehVenom/AvgMerge_Dolly-Pygmalion-6b\")\n", "}\n", - "model = model.get(Model, model)\n", + "model = models.get(Model)\n", "\n", "if StartKoboldAI:\n", " def downloadKobold():\n",