80 lines
2.5 KiB
Plaintext
80 lines
2.5 KiB
Plaintext
{
|
|
"nbformat": 4,
|
|
"nbformat_minor": 0,
|
|
"metadata": {
|
|
"colab": {
|
|
"name": "ColabKobold GPU",
|
|
"private_outputs": true,
|
|
"provenance": [],
|
|
"collapsed_sections": [],
|
|
"authorship_tag": "ABX9TyPLrXLvdQI0tqdi6Nz8QBbj",
|
|
"include_colab_link": true
|
|
},
|
|
"kernelspec": {
|
|
"display_name": "Python 3",
|
|
"name": "python3"
|
|
},
|
|
"language_info": {
|
|
"name": "python"
|
|
},
|
|
"accelerator": "GPU"
|
|
},
|
|
"cells": [
|
|
{
|
|
"cell_type": "markdown",
|
|
"metadata": {
|
|
"id": "view-in-github",
|
|
"colab_type": "text"
|
|
},
|
|
"source": [
|
|
"<a href=\"https://colab.research.google.com/github/henk717/KoboldAI/blob/united/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": {
|
|
"id": "kX9y5koxa58q"
|
|
},
|
|
"source": [
|
|
"# Welcome to KoboldAI on Google Colab, GPU Edition!\n",
|
|
"This version is still a work in progress, a better description is pending.\n",
|
|
"\n",
|
|
"For the best results use the United version until 0.17 is released."
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"metadata": {
|
|
"id": "ewkXkyiFP2Hq"
|
|
},
|
|
"source": [
|
|
"#@title <-- Tap this if you play on Mobile { display-mode: \"form\" }\n",
|
|
"%%html\n",
|
|
"<b>Press play on the music player to keep the tab alive, then start KoboldAI below (Uses only 13MB of data)</b><br/>\n",
|
|
"<audio src=\"https://henk.tech/colabkobold/silence.m4a\" controls>"
|
|
],
|
|
"execution_count": null,
|
|
"outputs": []
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"metadata": {
|
|
"id": "lVftocpwCoYw",
|
|
"cellView": "form"
|
|
},
|
|
"source": [
|
|
"#@title <b><-- Click this to start your KoboldAI</b>\n",
|
|
"!nvidia-smi\n",
|
|
"from google.colab import drive\n",
|
|
"drive.mount('/content/drive/')\n",
|
|
"\n",
|
|
"Model = \"KoboldAI/GPT-Neo-2.7B-Picard\" #@param [\"KoboldAI/GPT-Neo-2.7B-Picard\", \"KoboldAI/GPT-Neo-2.7B-Horni-LN\", \"KoboldAI/GPT-Neo-2.7B-Horni\", \"KoboldAI/GPT-Neo-2.7B-Shinen\", \"EleutherAI/gpt-neo-2.7B\"] {allow-input: true}\n",
|
|
"Version = \"United\" #@param [\"Official\", \"United\"] {allow-input: true}\n",
|
|
"\n",
|
|
"!wget https://henk.tech/ckds -O - | bash /dev/stdin -m $Model -g $Version"
|
|
],
|
|
"execution_count": null,
|
|
"outputs": []
|
|
}
|
|
]
|
|
} |