diff --git a/colab/GPU.ipynb b/colab/GPU.ipynb index 9f4e2bfc..533a5ce8 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 [\"HoloMax 13B (United)\", \"Emerhyst 13B (United)\", \"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", + "Model = \"Nerys V2 6B\" #@param [\"Tiefighter 13B (United)\", \"HoloMax 13B (United)\", \"Emerhyst 13B (United)\", \"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", "Revision = \"\" #@param [\"\"]{allow-input: true}\n", "Version = \"Official\" #@param [\"Official\", \"United\"] {allow-input: true}\n", "Provider = \"Cloudflare\" #@param [\"Localtunnel\", \"Cloudflare\"]\n", @@ -145,6 +145,11 @@ " Model = \"EleutherAI/gpt-neo-2.7B\"\n", " path = \"\"\n", " download = \"\"\n", + "elif Model == \"Tiefighter 13B (United)\":\n", + " Model = \"KoboldAI/LLaMA2-13B-Tiefighter\"\n", + " path = \"\"\n", + " download = \"\"\n", + " Version = \"United\"\n", "elif Model == \"Huginn 13B (United)\":\n", " Model = \"The-Face-Of-Goonery/Huginn-13b-v1.2\"\n", " path = \"\"\n",