diff --git a/colab/GPU.ipynb b/colab/GPU.ipynb index a2da0373a..c1474e479 100644 --- a/colab/GPU.ipynb +++ b/colab/GPU.ipynb @@ -65,7 +65,7 @@ "\n", "Model = \"Pygmalion 6B\" #@param [ \"Pygmalion 6B\", \"Pygmalion 6B Dev\"] {allow-input: true}\n", "Version = \"Official\" \n", - "KoboldAI_Provider = \"Cloudflare\" #@param [\"Localtunnel\", \"Cloudflare\"]\n", + "KoboldAI_Provider = \"Localtunnel\" #@param [\"Localtunnel\", \"Cloudflare\"]\n", "use_google_drive = True #@param {type:\"boolean\"}\n", "Provider = KoboldAI_Provider\n", "!nvidia-smi\n", @@ -133,7 +133,11 @@ "#Henk's KoboldAI script\n", "!wget https://koboldai.org/ckds && chmod +x ckds\n", "!./ckds --init only\n", - "p = subprocess.Popen(['/content/ckds', '--model', Model], stdout=subprocess.PIPE, stderr=subprocess.PIPE)\n", + "if Provider == \"Localtunnel\":\n", + " p = subprocess.Popen(['/content/ckds', '--model', Model, '--localtunnel', 'yes'], stdout=subprocess.PIPE, stderr=subprocess.PIPE)\n", + "else:\n", + " p = subprocess.Popen(['/content/ckds', '--model', Model], stdout=subprocess.PIPE, stderr=subprocess.PIPE)\n", + "\n", "\n", "#Do not repeat! Tricks performed by a professional!\n", "url = ''\n", diff --git a/public/index.html b/public/index.html index 296d22bf5..f82a2514d 100644 --- a/public/index.html +++ b/public/index.html @@ -393,6 +393,7 @@ $('#colab_shadow_popup').css('display', 'none'); is_colab = true; let url = String(getData.colaburl).split("flare.com")[0] + "flare.com"; + url = String(url).split("loca.lt")[0] + "loca.lt"; $('#api_url_text').val(url); setTimeout(function() { $('#api_button').click();