Allows to set the upscale factor for gfpgan face_enhancer

This commit is contained in:
Michael
2024-09-22 18:31:06 +01:00
parent aa021b6aa0
commit 07c735e9d2
3 changed files with 10 additions and 3 deletions
+4 -1
View File
@@ -71,6 +71,9 @@ def parse_args() -> None:
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('--enhancer_upscale_factor',
help='Sets the upscale factor for the enhancer. Only applies if `face_enhancer` is set as a frame_processor',
dest='enhancer_upscale_factor', type=int, default=1)
program.add_argument('-v', '--version', action='version',
version=f'{modules.metadata.name} {modules.metadata.version}')
@@ -100,7 +103,7 @@ def parse_args() -> None:
modules.globals.execution_providers = decode_execution_providers(args.execution_provider)
modules.globals.execution_threads = args.execution_threads
modules.globals.headless = args.headless
modules.globals.enhancer_upscale_factor = args.enhancer_upscale_factor
# Handle face enhancer tumbler
modules.globals.fp_ui['face_enhancer'] = 'face_enhancer' in args.frame_processor