From b66110ea54a0ca466c8aa1d14c0fa8e41626a89f Mon Sep 17 00:00:00 2001
From: henk717 <henk@henk.tech>
Date: Mon, 8 May 2023 18:54:41 +0200
Subject: [PATCH] Created using Colaboratory

---
 colab/TPU.ipynb | 19 +++++++++++++++++--
 1 file changed, 17 insertions(+), 2 deletions(-)

diff --git a/colab/TPU.ipynb b/colab/TPU.ipynb
index 20d0aba9..7fffcf74 100644
--- a/colab/TPU.ipynb
+++ b/colab/TPU.ipynb
@@ -1,5 +1,15 @@
 {
   "cells": [
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "view-in-github",
+        "colab_type": "text"
+      },
+      "source": [
+        "<a href=\"https://colab.research.google.com/github/KoboldAI/KoboldAI-Client/blob/main/colab/TPU.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
+      ]
+    },
     {
       "cell_type": "markdown",
       "source": [
@@ -56,7 +66,7 @@
         "#@title <b><-- Select your model below and then click this to start KoboldAI</b>\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\", \"Nerybus 13B\", \"Erebus 13B\", \"Janeway 13B\", \"Shinen 13B\", \"Skein 20B\", \"Erebus 20B\", \"Skein 6B\", \"Janeway 6B\", \"Adventure 6B\", \"Shinen 6B\", \"Lit V2 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\", \"Nerybus 13B\", \"Green Devil 13B\", \"Erebus 13B\", \"Janeway 13B\", \"Shinen 13B\", \"Skein 20B\", \"Erebus 20B\", \"Skein 6B\", \"Janeway 6B\", \"Adventure 6B\", \"Shinen 6B\", \"Lit V2 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",
         "use_google_drive = True #@param {type:\"boolean\"}\n",
@@ -93,6 +103,10 @@
         "  Model = \"KoboldAI/OPT-13B-Nerybus-Mix\"\n",
         "  path = \"\"\n",
         "  download = \"\"\n",
+        "elif Model == \"Green Devil 13B\":\n",
+        "  Model = \"Pirr/pythia-13b-deduped-green_devil\"\n",
+        "  path = \"\"\n",
+        "  download = \"\"\n",
         "elif Model == \"Erebus 13B\":\n",
         "  Model = \"KoboldAI/OPT-13B-Erebus\"\n",
         "  path = \"\"\n",
@@ -235,7 +249,8 @@
     "colab": {
       "name": "ColabKobold TPU",
       "provenance": [],
-      "private_outputs": true
+      "private_outputs": true,
+      "include_colab_link": true
     },
     "kernelspec": {
       "display_name": "Python 3",