diff --git a/app.py b/app.py index e85fa7da..641e8a4d 100644 --- a/app.py +++ b/app.py @@ -58,7 +58,7 @@ if __name__ == "__main__": if args.port: config.port=args.port - LOLLMSWebUI.build_instance(config=config, lollms_paths=lollms_paths, socketio=sio) + LOLLMSWebUI.build_instance(config=config, lollms_paths=lollms_paths, args=args, socketio=sio) lollmsElfServer:LOLLMSWebUI = LOLLMSWebUI.get_instance() # Import all endpoints from lollms.server.endpoints.lollms_binding_files_server import router as lollms_binding_files_server_router diff --git a/lollms_webui.py b/lollms_webui.py index ed3addeb..9222fce4 100644 --- a/lollms_webui.py +++ b/lollms_webui.py @@ -94,6 +94,7 @@ class LOLLMSWebUI(LOLLMSElfServer): try_select_binding=False, try_select_model=False, callback=None, + args=None, socketio = None ): if LOLLMSWebUI.__instance is None: @@ -107,6 +108,7 @@ class LOLLMSWebUI(LOLLMSElfServer): try_select_binding=try_select_binding, try_select_model=try_select_model, callback=callback, + args=args, socketio=socketio ) return LOLLMSWebUI.__instance @@ -121,6 +123,7 @@ class LOLLMSWebUI(LOLLMSElfServer): try_select_binding=False, try_select_model=False, callback=None, + args=None, socketio=None ) -> None: super().__init__( @@ -137,6 +140,7 @@ class LOLLMSWebUI(LOLLMSElfServer): ) self.app_name:str = "LOLLMSWebUI" self.version:str = lollms_webui_version + self.args = args self.busy = False