From 66264d38c416e6c4c79d227b5bd6fb5ef9ad4b70 Mon Sep 17 00:00:00 2001 From: henk717 Date: Tue, 28 Mar 2023 00:23:10 +0200 Subject: [PATCH] Add Mixes --- colab/GPU.ipynb | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/colab/GPU.ipynb b/colab/GPU.ipynb index 5e088809..7d9fbb5f 100644 --- a/colab/GPU.ipynb +++ b/colab/GPU.ipynb @@ -82,7 +82,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 V2 6B\" #@param [\"Nerys V2 6B\", \"Erebus 6B\", \"Skein 6B\", \"Janeway 6B\", \"Adventure 6B\", \"Pygmalion 6B\", \"Pygmalion 6B Dev\", \"Lit V2 6B\", \"Lit 6B\", \"Shinen 6B\", \"Nerys 2.7B\", \"AID 2.7B\", \"Erebus 2.7B\", \"Janeway 2.7B\", \"Picard 2.7B\", \"Horni LN 2.7B\", \"Horni 2.7B\", \"Shinen 2.7B\", \"OPT 2.7B\", \"Fairseq Dense 2.7B\", \"Neo 2.7B\"] {allow-input: true}\n", + "Model = \"Nerys V2 6B\" #@param [\"Nerys V2 6B\", \"Nerybus 6B\", \"Erebus 6B\", \"Skein 6B\", \"Janeway 6B\", \"Adventure 6B\", \"PPO_Pygway 6B\", \"Pygmalion 6B\", \"Pygmalion 6B Dev\", \"Lit V2 6B\", \"Lit 6B\", \"Shinen 6B\", \"Nerys 2.7B\", \"AID 2.7B\", \"Erebus 2.7B\", \"Janeway 2.7B\", \"Picard 2.7B\", \"Horni LN 2.7B\", \"Horni 2.7B\", \"Shinen 2.7B\", \"OPT 2.7B\", \"Fairseq Dense 2.7B\", \"Neo 2.7B\"] {allow-input: true}\n", "Version = \"Official\" #@param [\"Official\", \"United\"] {allow-input: true}\n", "Provider = \"Cloudflare\" #@param [\"Localtunnel\", \"Cloudflare\"]\n", "use_google_drive = True #@param {type:\"boolean\"}\n", @@ -104,6 +104,10 @@ " Model = \"KoboldAI/OPT-6B-nerys-v2\"\n", " path = \"\"\n", " download = \"\"\n", + "elif Model == \"Nerybus 6B\":\n", + " Model = \"KoboldAI/OPT-6.7B-Nerybus-Mix\"\n", + " path = \"\"\n", + " download = \"\"\n", "elif Model == \"Erebus 6B\":\n", " Model = \"KoboldAI/OPT-6.7B-Erebus\"\n", " path = \"\"\n", @@ -120,6 +124,11 @@ " Model = \"KoboldAI/GPT-J-6B-Adventure\"\n", " path = \"\"\n", " download = \"\"\n", + "elif Model == \"PPO_Pygway 6B\":\n", + " Model = \"KoboldAI/PPO_Pygway-6b-Mix\"\n", + " path = \"\"\n", + " download = \"\"\n", + " Version = \"United\"\n", "elif Model == \"Pygmalion 6B\":\n", " Model = \"PygmalionAI/pygmalion-6b\"\n", " path = \"\"\n",