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

Model for cross-module globals

Browse Source
This commit is contained in:
SillyLossy
2023-04-07 15:23:41 +03:00
parent ec7a839324
commit ed3324fadc
3 changed files with 15 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
colab/GPU.ipynb
View File

@@ -245,9 +245,10 @@
"\n",
"# ---\n",
"# TavernAI extras\n",
"extras_url = '(disabled)'\n",
"params = []\n",
"params.append('--cpu')\n",
"import globals\n",
"globals.extras_url = '(disabled)'\n",
"globals.params = []\n",
"globals.params.append('--cpu')\n",
"ExtrasModules = []\n",
"\n",
"if (extras_enable_captioning):\n",
@@ -257,10 +258,10 @@
"if (extras_enable_emotions):\n",
" ExtrasModules.append('classify')\n",
"\n",
"params.append(f'--classification-model={Emotions_Model}')\n",
"params.append(f'--summarization-model={Memory_Model}')\n",
"params.append(f'--captioning-model={Captions_Model}')\n",
"params.append(f'--enable-modules={\",\".join(ExtrasModules)}')\n",
"globals.params.append(f'--classification-model={Emotions_Model}')\n",
"globals.params.append(f'--summarization-model={Memory_Model}')\n",
"globals.params.append(f'--captioning-model={Captions_Model}')\n",
"globals.params.append(f'--enable-modules={\",\".join(ExtrasModules)}')\n",
"\n",
"\n",
"if UseExtrasExtensions:\n",
@@ -315,7 +316,7 @@
"\n",
"%env colaburl=$url\n",
"%env SILLY_TAVERN_PORT=5001\n",
"print(\"KoboldAI LINK:\", url, '###Extensions API LINK###', extras_url, \"###SillyTavern LINK###\", sep=\"\\n\")\n",
"print(\"KoboldAI LINK:\", url, '###Extensions API LINK###', globals.extras_url, \"###SillyTavern LINK###\", sep=\"\\n\")\n",
"p = subprocess.Popen([\"lt\", \"--port\", \"5001\"], stdout=subprocess.PIPE, stderr=subprocess.PIPE)\n",
"print(p.stdout.readline().decode().strip())\n",
"!node server.js"