better gradio args for win

close #18
This commit is contained in:
layerdiffusion 2025-04-17 11:20:36 -07:00
parent 0e5fe5d7ca
commit 1b371ad792

View File

@ -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,
)