Streamlining Revision Support

colab/GPU.ipynb

@@ -5,8 +5,7 @@
     "colab": {
       "name": "ColabKobold GPU",
       "private_outputs": true,
-      "provenance": [],
-      "include_colab_link": true
+      "provenance": []
     },
     "kernelspec": {
       "display_name": "Python 3",
@@ -18,16 +17,6 @@
     "accelerator": "GPU"
   },
   "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/GPU.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
-      ]
-    },
     {
       "cell_type": "markdown",
       "metadata": {
@@ -124,10 +113,12 @@
         "  Model = \"PygmalionAI/pygmalion-6b\"\n",
         "  path = \"\"\n",
         "  download = \"\"\n",
+        "  Version = \"United\"\n",
         "elif Model == \"Pygmalion 6B Dev\":\n",
         "  Model = \"PygmalionAI/pygmalion-6b\"\n",
-        "  Revision = \"dev\"\n",
+        "  Revision = \"--revision dev\"\n",
         "  path = \"\"\n",
+        "  Version = \"United\"\n",
         "  download = \"\"\n",
         "elif Model == \"Lit V2 6B\":\n",
         "  Model = \"hakurei/litv2-6B-rev3\"\n",
@@ -191,7 +182,7 @@
         "else:\n",
         "  tunnel = \"\"\n",
         "\n",
-        "!wget https://koboldai.org/ckds -O - | bash /dev/stdin -m $Model -g $Version $tunnel -r $Revision"
+        "!wget https://koboldai.org/ckds -O - | bash /dev/stdin -m $Model -g $Version $Revision $tunnel"
       ],
       "execution_count": null,
       "outputs": []