--noaimenu as seperate parameter

This commit is contained in:
Henk 2022-06-07 20:44:14 +02:00
parent 2333c85f4e
commit 66ba165b4c
1 changed files with 3 additions and 1 deletions

View File

@ -864,6 +864,7 @@ def general_startup():
# Parsing Parameters # Parsing Parameters
parser = argparse.ArgumentParser(description="KoboldAI Server") parser = argparse.ArgumentParser(description="KoboldAI Server")
parser.add_argument("--remote", action='store_true', help="Optimizes KoboldAI for Remote Play") parser.add_argument("--remote", action='store_true', help="Optimizes KoboldAI for Remote Play")
parser.add_argument("--noaimenu", action='store_true', help="Disables the ability to select the AI")
parser.add_argument("--ngrok", action='store_true', help="Optimizes KoboldAI for Remote Play using Ngrok") parser.add_argument("--ngrok", action='store_true', help="Optimizes KoboldAI for Remote Play using Ngrok")
parser.add_argument("--localtunnel", action='store_true', help="Optimizes KoboldAI for Remote Play using Localtunnel") parser.add_argument("--localtunnel", action='store_true', help="Optimizes KoboldAI for Remote Play using Localtunnel")
parser.add_argument("--host", action='store_true', help="Optimizes KoboldAI for Remote Play without using a proxy service") parser.add_argument("--host", action='store_true', help="Optimizes KoboldAI for Remote Play without using a proxy service")
@ -903,6 +904,7 @@ def general_startup():
args.nobreakmodel = True; args.nobreakmodel = True;
args.quiet = True; args.quiet = True;
args.lowmem = True; args.lowmem = True;
args.noaimenu = True;
if args.quiet: if args.quiet:
vars.quiet = True vars.quiet = True
@ -1936,7 +1938,7 @@ def load_model(use_gpu=True, gpu_layers=None, initial_load=False, online_model="
@app.route('/') @app.route('/')
@app.route('/index') @app.route('/index')
def index(): def index():
return render_template('index.html', hide_ai_menu=args.remote) return render_template('index.html', hide_ai_menu=args.noaimenu)
@app.route('/download') @app.route('/download')
def download(): def download():
save_format = request.args.get("format", "json").strip().lower() save_format = request.args.get("format", "json").strip().lower()