rixty/SillyTavern
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern.git synced 2025-06-05 21:59:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

Colab again

Browse Source
This commit is contained in:
SillyLossy
2023-04-06 22:23:31 +03:00
parent 1a3f9dc10f
commit f9d83512e0
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
colab/GPU.ipynb
View File

@@ -181,7 +181,7 @@
" \"/content/drive/MyDrive\"\n", " \"/content/drive/MyDrive\"\n",
" ])\n", " ])\n",
"\n", "\n",
"model = {\n", "models = {\n",
" \"Nerys V2 6B\": ModelData(\"KoboldAI/OPT-6B-nerys-v2\"),\n", " \"Nerys V2 6B\": ModelData(\"KoboldAI/OPT-6B-nerys-v2\"),\n",
" \"Erebus 6B\": ModelData(\"KoboldAI/OPT-6.7B-Erebus\"),\n", " \"Erebus 6B\": ModelData(\"KoboldAI/OPT-6.7B-Erebus\"),\n",
" \"Skein 6B\": ModelData(\"KoboldAI/GPT-J-6B-Skein\"),\n", " \"Skein 6B\": ModelData(\"KoboldAI/GPT-J-6B-Skein\"),\n",
@@ -226,7 +226,7 @@
" \"Dolly_GPT-J-6b\": ModelData(\"TehVenom/Dolly_GPT-J-6b\"),\n", " \"Dolly_GPT-J-6b\": ModelData(\"TehVenom/Dolly_GPT-J-6b\"),\n",
" \"Dolly_Pyg-6B\": ModelData(\"TehVenom/AvgMerge_Dolly-Pygmalion-6b\")\n", " \"Dolly_Pyg-6B\": ModelData(\"TehVenom/AvgMerge_Dolly-Pygmalion-6b\")\n",
"}\n", "}\n",
"model = model.get(Model, model)\n", "model = models.get(Model)\n",
"\n", "\n",
"if StartKoboldAI:\n", "if StartKoboldAI:\n",
" def downloadKobold():\n", " def downloadKobold():\n",