Spaces:
Running
on
A100
Running
on
A100
store_true for bool args
Browse files
config.py
CHANGED
@@ -47,21 +47,21 @@ parser.add_argument("--timeout", type=float, default=TIMEOUT, help="Timeout")
|
|
47 |
parser.add_argument(
|
48 |
"--safety-checker",
|
49 |
"--safety_checker",
|
50 |
-
|
51 |
default=SAFETY_CHECKER,
|
52 |
help="Safety Checker",
|
53 |
)
|
54 |
parser.add_argument(
|
55 |
"--torch-compile",
|
56 |
"--torch_compile",
|
57 |
-
|
58 |
default=TORCH_COMPILE,
|
59 |
help="Torch Compile",
|
60 |
)
|
61 |
parser.add_argument(
|
62 |
"--use-taesd",
|
63 |
"--use_taesd",
|
64 |
-
|
65 |
default=USE_TAESD,
|
66 |
help="Use Tiny Autoencoder",
|
67 |
)
|
@@ -87,13 +87,13 @@ parser.add_argument(
|
|
87 |
)
|
88 |
parser.add_argument(
|
89 |
"--debug",
|
90 |
-
|
91 |
default=False,
|
92 |
help="Debug",
|
93 |
)
|
94 |
parser.add_argument(
|
95 |
"--compel",
|
96 |
-
|
97 |
default=False,
|
98 |
help="Compel",
|
99 |
)
|
|
|
47 |
parser.add_argument(
|
48 |
"--safety-checker",
|
49 |
"--safety_checker",
|
50 |
+
action="store_true",
|
51 |
default=SAFETY_CHECKER,
|
52 |
help="Safety Checker",
|
53 |
)
|
54 |
parser.add_argument(
|
55 |
"--torch-compile",
|
56 |
"--torch_compile",
|
57 |
+
action="store_true",
|
58 |
default=TORCH_COMPILE,
|
59 |
help="Torch Compile",
|
60 |
)
|
61 |
parser.add_argument(
|
62 |
"--use-taesd",
|
63 |
"--use_taesd",
|
64 |
+
action="store_true",
|
65 |
default=USE_TAESD,
|
66 |
help="Use Tiny Autoencoder",
|
67 |
)
|
|
|
87 |
)
|
88 |
parser.add_argument(
|
89 |
"--debug",
|
90 |
+
action="store_true",
|
91 |
default=False,
|
92 |
help="Debug",
|
93 |
)
|
94 |
parser.add_argument(
|
95 |
"--compel",
|
96 |
+
action="store_true",
|
97 |
default=False,
|
98 |
help="Compel",
|
99 |
)
|