Add a --listen argument to listen on 0.0.0.0

43c795f4 · comfyanonymous · 41a7532c · 43c795f4
Commit 43c795f4 authored Feb 04, 2023 by comfyanonymous
Show whitespace changes
Inline Side-by-side

Showing with 5 additions and 1 deletion

main.py main.py +5 -1

No files found.
--- a/main.py
+++ b/main.py
@@ -445,6 +445,10 @@ def run(prompt_queue, address='', port=8188):
 if __name__ == "__main__":
    q = PromptQueue()
    threading.Thread(target=prompt_worker, daemon=True, args=(q,)).start()
-    run(q, address='127.0.0.1', port=8188)
+    if '--listen' in sys.argv:
+        address = '0.0.0.0'
+    else:
+        address = '127.0.0.1'
+    run(q, address=address, port=8188)