From 31dd6c0531367c40537e1fc866db31ac842cd5f4 Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Tue, 28 Mar 2023 01:42:34 -0400 Subject: [PATCH] Add way to specify listen ip with --listen. --- main.py | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/main.py b/main.py index 26bad1b8..d82d1d6e 100644 --- a/main.py +++ b/main.py @@ -12,7 +12,7 @@ if os.name == "nt": if __name__ == "__main__": if '--help' in sys.argv: print("Valid Command line Arguments:") - print("\t--listen\t\t\tListen on 0.0.0.0 so the UI can be accessed from other computers.") + print("\t--listen [ip]\t\t\tListen on ip or 0.0.0.0 if none given so the UI can be accessed from other computers.") print("\t--port 8188\t\t\tSet the listen port.") print("\t--dont-upcast-attention\t\tDisable upcasting of attention \n\t\t\t\t\tcan boost speed but increase the chances of black images.\n") print("\t--use-split-cross-attention\tUse the split cross attention optimization instead of the sub-quadratic one.\n\t\t\t\t\tIgnored when xformers is used.") @@ -92,11 +92,19 @@ if __name__ == "__main__": hijack_progress(server) threading.Thread(target=prompt_worker, daemon=True, args=(q,server,)).start() - if '--listen' in sys.argv: + try: address = '0.0.0.0' - else: + p_index = sys.argv.index('--listen') + try: + ip = sys.argv[p_index + 1] + if ip[:2] != '--': + address = ip + except: + pass + except: address = '127.0.0.1' + dont_print = False if '--dont-print-server' in sys.argv: dont_print = True