From 1b371ad7923355f85bac5d904a138961fb19ea30 Mon Sep 17 00:00:00 2001 From: layerdiffusion <19834515+lllyasviel@users.noreply.github.com> Date: Thu, 17 Apr 2025 11:20:36 -0700 Subject: [PATCH] better gradio args for win close #18 --- demo_gradio.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/demo_gradio.py b/demo_gradio.py index 632cb87..687cf56 100644 --- a/demo_gradio.py +++ b/demo_gradio.py @@ -31,9 +31,13 @@ from diffusers_helper.bucket_tools import find_nearest_bucket parser = argparse.ArgumentParser() parser.add_argument('--share', action='store_true') parser.add_argument("--server", type=str, default='0.0.0.0') -parser.add_argument("--port", type=int, default=7860) +parser.add_argument("--port", type=int, required=False) +parser.add_argument("--inbrowser", action='store_true') args = parser.parse_args() +# for win desktop probably use --server 127.0.0.1 --inbrowser +# For linux server probably use --server 127.0.0.1 or do not use any cmd flags + print(args) free_mem_gb = get_cuda_free_memory_gb(gpu) @@ -396,4 +400,5 @@ block.launch( server_name=args.server, server_port=args.port, share=args.share, + inbrowser=args.inbrowser, )