You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
16 lines
456 B
Python
16 lines
456 B
Python
from argparse import ArgumentParser
|
|
|
|
|
|
def preload(parser: ArgumentParser) -> None:
|
|
parser.add_argument(
|
|
"--faceswaplab_loglevel",
|
|
default="INFO",
|
|
choices=["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"],
|
|
help="Set the log level (DEBUG, INFO, WARNING, ERROR, CRITICAL)",
|
|
)
|
|
parser.add_argument(
|
|
"--faceswaplab_gpu",
|
|
action="store_true",
|
|
help="Enable GPU if set, disable if not set",
|
|
)
|