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

Potential Colab Fix

Browse Source
This commit is contained in:
ebolam
2022-10-28 19:19:58 -04:00
parent 0aa558f17b
commit f48d5efac1
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
aiserver.py
View File

@@ -12219,7 +12219,7 @@ def run():
logger.init_ok("Flask", status="OK") logger.init_ok("Flask", status="OK")
logger.init("Webserver", status="Starting") logger.init("Webserver", status="Starting")
patch_transformers() patch_transformers()
startup()
# Start Flask/SocketIO (Blocking, so this must be last method!) # Start Flask/SocketIO (Blocking, so this must be last method!)
port = args.port if "port" in args and args.port is not None else 5000 port = args.port if "port" in args and args.port is not None else 5000
koboldai_vars.port = port koboldai_vars.port = port
@@ -12252,6 +12252,9 @@ def run():
from flask_cloudflared import _run_cloudflared from flask_cloudflared import _run_cloudflared
cloudflare = _run_cloudflared(port) cloudflare = _run_cloudflared(port)
koboldai_vars.cloudflare_link = cloudflare koboldai_vars.cloudflare_link = cloudflare
startup()
if(args.localtunnel or args.ngrok or args.remote): if(args.localtunnel or args.ngrok or args.remote):
with open('cloudflare.log', 'w') as cloudflarelog: with open('cloudflare.log', 'w') as cloudflarelog:
cloudflarelog.write("KoboldAI has finished loading and is available at the following link : " + cloudflare) cloudflarelog.write("KoboldAI has finished loading and is available at the following link : " + cloudflare)