diff --git a/modules/core.py b/modules/core.py index 55707e2..b83a3ef 100644 --- a/modules/core.py +++ b/modules/core.py @@ -48,6 +48,7 @@ def parse_args() -> None: program.add_argument('--max-memory', help='maximum amount of RAM in GB', dest='max_memory', type=int, default=suggest_max_memory()) program.add_argument('--execution-provider', help='execution provider', dest='execution_provider', default=['cpu'], choices=suggest_execution_providers(), nargs='+') program.add_argument('--execution-threads', help='number of execution threads', dest='execution_threads', type=int, default=suggest_execution_threads()) + program.add_argument('--headless', help='Run in headless mode', dest='headless', default=False, action='store_true') program.add_argument('-v', '--version', action='version', version=f'{modules.metadata.name} {modules.metadata.version}') # register deprecated args @@ -76,6 +77,7 @@ def parse_args() -> None: modules.globals.max_memory = args.max_memory modules.globals.execution_providers = decode_execution_providers(args.execution_provider) modules.globals.execution_threads = args.execution_threads + modules.globals.headless = args.headless #for ENHANCER tumbler: if 'face_enhancer' in args.frame_processor: