diff --git a/colab/GPU.ipynb b/colab/GPU.ipynb index d7e5f462f..80adbbc97 100644 --- a/colab/GPU.ipynb +++ b/colab/GPU.ipynb @@ -314,20 +314,13 @@ "\n", "%env colaburl=$url\n", "%env SILLY_TAVERN_PORT=5001\n", - "from flask_cloudflared import start_cloudflared\n", "!sed -i 's/listen = true/listen = false/g' config.conf\n", "!touch stdout.log stderr.log\n", "!forever start -o stdout.log -e stderr.log server.js\n", "print(\"KoboldAI LINK:\", url, '###Extensions API LINK###', globals.extras_url, \"###SillyTavern LINK###\", sep=\"\\n\")\n", - "import inspect\n", - "import random\n", - "sig = inspect.signature(start_cloudflared)\n", - "sum = sum(1 for param in sig.parameters.values() if param.kind == param.POSITIONAL_OR_KEYWORD)\n", - "if sum > 1:\n", - " metrics_port = random.randint(8100, 9000)\n", - " start_cloudflared(5001, metrics_port)\n", - "else:\n", - " start_cloudflared(5001)\n", + "from flask_cloudflared import _run_cloudflared\n", + "cloudflare = _run_cloudflared(5001)\n", + "print(cloudflare)\n", "!tail -f stdout.log stderr.log" ] }