Add --port to set custom port.

1f18221e · comfyanonymous · 6e40393b · 1f18221e
Commit 1f18221e authored Feb 07, 2023 by comfyanonymous
Show whitespace changes
Inline Side-by-side

Showing with 9 additions and 1 deletion

main.py main.py +9 -1

No files found.
--- a/main.py
+++ b/main.py
@@ -449,6 +449,14 @@ if __name__ == "__main__":
        address = '0.0.0.0'
    else:
        address = '127.0.0.1'
-    run(q, address=address, port=8188)
+
+    port = 8188
+    try:
+        p_index = sys.argv.index('--port')
+        port = int(sys.argv[p_index + 1])
+    except:
+        pass
+
+    run(q, address=address, port=port)