From e3e65947f2e2cf32dad424e2f3ad33481b28ad28 Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Tue, 7 Feb 2023 22:12:56 -0500 Subject: [PATCH] Add a --help to main.py --- main.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/main.py b/main.py index 092df660..4f7bb65b 100644 --- a/main.py +++ b/main.py @@ -6,6 +6,16 @@ import threading import heapq import traceback +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--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.") + print() + exit() + if '--dont-upcast-attention' in sys.argv: print("disabling upcasting of attention") os.environ['ATTN_PRECISION'] = "fp16"