diff --git a/colab/GPU.ipynb b/colab/GPU.ipynb index c1474e479..a4accca54 100644 --- a/colab/GPU.ipynb +++ b/colab/GPU.ipynb @@ -1,192 +1,413 @@ { "cells": [ { + "attachments": {}, "cell_type": "markdown", - "metadata": { - "id": "d-Yihz3hAb2E" - }, + "metadata": {}, "source": [ - "https://colab.research.google.com/github/TavernAI/TavernAI/blob/main/colab/GPU.ipynb
\n", - "\n", - "Works with:
\n", - "KoboldAI https://github.com/KoboldAI/KoboldAI-Client
\n", - "Pygmalion https://huggingface.co/PygmalionAI/
\n", - "
\n", "**Links**
\n", - "TavernAI Github https://github.com/TavernAI/TavernAI
\n", + "Cohee's TavernAI fork Github https://github.com/Cohee1207/SillyTavern
\n", + "Cohee's TavernAI Extras Github https://github.com/Cohee1207/TavernAI-extras/
\n", "TavernAI Discord https://discord.gg/zmK2gmr45t
\n", - "TavernAI Boosty https://boosty.to/tavernai\n", - "
\n",
-        " Tavern.AI/ \\ /  ^   ^ ^ ^    ~~~~ ^ \\     /  ^ ^   ^ ^/ ^  ^ \\/^  ^    \\\n",
-        "         /^ ^\\ ^  ^ ^   ^ ^  ~~   ^   \\   /  ^  ^ ^   / ^ ^  ^/   ^ ^    \\\n",
-        "        /^ ^ ^\\^   ^ ^ ^   _||____   ^ \\ /  ^  ^ ^   /       /  ^  ^  ^   \\\n",
-        " /\\ /\\ /\\   ^  \\  /\\ /\\   /\\\\\\\\\\\\\\\\   ^ \\  ^ /\\ /\\ /\\   /\\ /\\ /\\  ^ ^  ^/\\\n",
-        "//\\\\/\\\\/\\\\   ^  \\//\\\\/\\\\ /__\\\\\\\\\\\\\\\\  _, \\  //\\\\/\\\\/\\\\ //\\\\/\\\\/\\\\  ^ ^ //\\\\\n",
-        "//\\\\/\\\\/\\\\       //\\\\/\\\\ |__|_|_|__|   \\__, //\\\\/\\\\/\\\\ //\\\\/\\\\/\\\\     ///\\\\\\\n",
-        " || || (@^◡^)(≖ ‸ ≖*) ( ←_← )\\| /|   /\\ \\ヽ(°ㅂ°╬) |( Ψ▼ー▼)∈ (O_O; )  |||\n",
-        "~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ ~~~~~ ~~~~~ ~~~~~ ~~~~~  ~~~~~ ~~ \n",
-        "
\n", - "**Launch Instructions**
\n", - "1. Click the launch button.\n", - "2. Wait for the environment and model to load\n", - "3. After initialization, a TavernAI link will appear\n", - "\n", - "**Faq**
\n", - "* Q: I do not get a TavernAI link\n", - "* A: It seems the localtunnel service is currently down, so the TavernAI link is unavailable. Need to wait for it to start working again." + "Questions? Hit me up on Discord: Cohee#1207" ] }, { "cell_type": "code", "execution_count": null, "metadata": { - "id": "hCpoIHxYcDGs" + "cellView": "form", + "id": "_1gpebrnlp5-" + }, + "outputs": [], + "source": [ + "#@title <-- Convert TavernAI characters to SillyTavern format\n", + "\n", + "!mkdir /convert\n", + "%cd /convert\n", + "\n", + "import os\n", + "from google.colab import drive\n", + "\n", + "drive.mount(\"/convert/drive\")\n", + "\n", + "!git clone -b tools https://github.com/EnergoStalin/SillyTavern.git\n", + "%cd SillyTavern\n", + "\n", + "!curl -o- https://raw.githubusercontent.com/nvm-sh/nvm/v0.37.2/install.sh | bash\n", + "!nvm install 19.1.0\n", + "!nvm use 19.1.0\n", + "\n", + "%cd tools/charaverter\n", + "\n", + "!npm i\n", + "\n", + "path = \"/convert/drive/MyDrive/TavernAI/characters\"\n", + "output = \"/convert/drive/MyDrive/SillyTavern/characters\"\n", + "if not os.path.exists(path):\n", + " path = output\n", + "\n", + "!mkdir -p $output\n", + "!node main.mjs $path $output\n", + "\n", + "drive.flush_and_unmount()\n", + "\n", + "%cd /\n", + "!rm -rf /convert" + ] + }, + { + "cell_type": "code", + "execution_count": null, + "metadata": { + "id": "ewkXkyiFP2Hq" }, "outputs": [], "source": [ "#@title <-- Tap this if you play on Mobile { display-mode: \"form\" }\n", - "#Taken from KoboldAI colab\n", "%%html\n", - "Press play on the music player to keep the tab alive, then start TavernAI below (Uses only 13MB of data)
\n", - "