From c55d1503ee2058aa7a63515dcfb02e86f8b5d131 Mon Sep 17 00:00:00 2001 From: henk717 Date: Fri, 19 Aug 2022 17:23:11 +0200 Subject: [PATCH] Skein 20B --- colab/TPU.ipynb | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/colab/TPU.ipynb b/colab/TPU.ipynb index 448880e7..22c35e08 100644 --- a/colab/TPU.ipynb +++ b/colab/TPU.ipynb @@ -66,7 +66,7 @@ "#@title <-- Select your model below and then click this to start KoboldAI\n", "#@markdown You can find a description of the models below along with instructions on how to start KoboldAI.\n", "\n", - "Model = \"Nerys 13B V2\" #@param [\"Nerys 13B V2\", \"Janeway 13B\", \"Shinen 13B\", \"Skein 6B\", \"Janeway 6B\", \"Adventure 6B\", \"Shinen 6B\", \"Lit 6B\", \"NeoX 20B\", \"OPT 13B\", \"Fairseq Dense 13B\", \"GPT-J-6B\"] {allow-input: true}\n", + "Model = \"Nerys 13B V2\" #@param [\"Nerys 13B V2\", \"Janeway 13B\", \"Shinen 13B\", \"Skein 20B\", \"Skein 6B\", \"Janeway 6B\", \"Adventure 6B\", \"Shinen 6B\", \"Lit 6B\", \"NeoX 20B\", \"OPT 13B\", \"Fairseq Dense 13B\", \"GPT-J-6B\"] {allow-input: true}\n", "Version = \"Official\" #@param [\"Official\", \"United\"] {allow-input: true}\n", "Provider = \"Cloudflare\" #@param [\"Localtunnel\", \"Cloudflare\"]\n", "\n", @@ -93,6 +93,10 @@ " Model = \"KoboldAI/fairseq-dense-13B-Shinen\"\n", " path = \"\"\n", " download = \"\"\n", + "elif Model == \"Skein 20B\":\n", + " Model = \"KoboldAI/GPT-NeoX-20B-Skein\"\n", + " path = \"\"\n", + " download = \"\"\n", "elif Model == \"NeoX 20B\":\n", " Model = \"EleutherAI/gpt-neox-20b\"\n", " path = \"\"\n", @@ -128,7 +132,7 @@ "elif Model == \"GPT-J-6B\":\n", " Model = \"EleutherAI/gpt-j-6B\"\n", " path = \"\"\n", - " download = \"\"\n", + " download = \"\"\n", "else:\n", " path = \"\"\n", " download = \"\"\n", @@ -225,4 +229,4 @@ }, "nbformat": 4, "nbformat_minor": 0 -} +} \ No newline at end of file