diff --git a/aiserver.py b/aiserver.py index 54346eb9..e0bb2160 100644 --- a/aiserver.py +++ b/aiserver.py @@ -864,6 +864,7 @@ def general_startup(): # Parsing Parameters parser = argparse.ArgumentParser(description="KoboldAI Server") 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("--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") @@ -903,6 +904,7 @@ def general_startup(): args.nobreakmodel = True; args.quiet = True; args.lowmem = True; + args.noaimenu = True; if args.quiet: 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('/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') def download(): save_format = request.args.get("format", "json").strip().lower()