From 9cb93d6b4c845a830228fda8c58749da22b7db20 Mon Sep 17 00:00:00 2001 From: henk717 Date: Thu, 10 Aug 2023 23:56:44 +0200 Subject: [PATCH] Add some 13B's for easier beta testing --- colab/GPU.ipynb | 32 +++++++++++++++++++++++++++++++- 1 file changed, 31 insertions(+), 1 deletion(-) diff --git a/colab/GPU.ipynb b/colab/GPU.ipynb index 78219b06..1835ce0e 100644 --- a/colab/GPU.ipynb +++ b/colab/GPU.ipynb @@ -80,7 +80,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\", \"Skein 6B\", \"Janeway 6B\", \"Adventure 6B\", \"Nerys 2.7B\", \"AID 2.7B\", \"Janeway 2.7B\", \"Picard 2.7B\", \"OPT 2.7B\", \"Fairseq Dense 2.7B\", \"Neo 2.7B\"] {allow-input: true}\n", + "Model = \"Nerys V2 6B\" #@param [\"MythoMax 13B (United)\", \"Huginn 13B (United)\", \"Chronos 13B (United)\", \"Airoboros M2.0 13B (United)\", \"Holodeck 13B (United)\", \"Spring Dragon 13B (United)\", \"Nerys V2 6B\", \"Skein 6B\", \"Janeway 6B\", \"Adventure 6B\", \"Nerys 2.7B\", \"AID 2.7B\", \"Janeway 2.7B\", \"Picard 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", @@ -146,6 +146,36 @@ " Model = \"EleutherAI/gpt-neo-2.7B\"\n", " path = \"\"\n", " download = \"\"\n", + "elif Model == \"Huginn 13B (United)\":\n", + " Model = \"The-Face-Of-Goonery/Huginn-13b-v1.2\"\n", + " path = \"\"\n", + " download = \"\"\n", + " Version = \"United\"\n", + "elif Model == \"Chronos 13B (United)\":\n", + " Model = \"elinas/chronos-13b-v2\"\n", + " path = \"\"\n", + " download = \"\"\n", + " Version = \"United\"\n", + "elif Model == \"Airoboros M2.0 13B (United)\":\n", + " Model = \"jondurbin/airoboros-l2-13b-gpt4-m2.0\"\n", + " path = \"\"\n", + " download = \"\"\n", + " Version = \"United\"\n", + "elif Model == \"MythoMax 13B (United)\":\n", + " Model = \"Gryphe/MythoMax-L2-13b\"\n", + " path = \"\"\n", + " download = \"\"\n", + " Version = \"United\"\n", + "elif Model == \"Spring Dragon 13B (United)\":\n", + " Model = \"Henk717/spring-dragon\"\n", + " path = \"\"\n", + " download = \"\"\n", + " Version = \"United\"\n", + "elif Model == \"Holodeck 13B (United)\":\n", + " Model = \"KoboldAI/LLAMA2-13B-Holodeck-1\"\n", + " path = \"\"\n", + " download = \"\"\n", + " Version = \"United\"\n", "\n", "if Provider == \"Localtunnel\":\n", " tunnel = \"--localtunnel yes\"\n",