fix(api): supply missing parameter to GFPGAN
This commit is contained in:
parent
ec1e3614f5
commit
d5e5588939
|
@ -24,7 +24,7 @@ last_pipeline_instance = None
|
||||||
last_pipeline_params = None
|
last_pipeline_params = None
|
||||||
|
|
||||||
|
|
||||||
def load_gfpgan(ctx: ServerContext, upscale: UpscaleParams):
|
def load_gfpgan(ctx: ServerContext, upscale: UpscaleParams, upsampler: Optional[RealESRGANer] = None):
|
||||||
global last_pipeline_instance
|
global last_pipeline_instance
|
||||||
global last_pipeline_params
|
global last_pipeline_params
|
||||||
|
|
||||||
|
@ -67,7 +67,7 @@ def correct_gfpgan(
|
||||||
return image
|
return image
|
||||||
|
|
||||||
logger.info('correcting faces with GFPGAN model: %s', upscale.correction_model)
|
logger.info('correcting faces with GFPGAN model: %s', upscale.correction_model)
|
||||||
gfpgan = load_gfpgan(ctx, upscale)
|
gfpgan = load_gfpgan(ctx, upscale, upsampler=upsampler)
|
||||||
|
|
||||||
_, _, output = gfpgan.enhance(
|
_, _, output = gfpgan.enhance(
|
||||||
image, has_aligned=False, only_center_face=False, paste_back=True, weight=upscale.face_strength)
|
image, has_aligned=False, only_center_face=False, paste_back=True, weight=upscale.face_strength)
|
||||||
|
|
Loading…
Reference in New Issue