apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): make stage-specific source image optional in GFPGAN stage (#233)

This commit is contained in:
Sean Sube 2023-03-10 18:15:31 -06:00
parent da3d95ff49
commit 6bdaa4a4ed
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
api/onnx_web/chain/correct_gfpgan.py
View File

@ -1,5 +1,6 @@
from logging import getLogger
from os import path
from typing import Optional
import numpy as np
from PIL import Image
@ -54,7 +55,7 @@ def correct_gfpgan(
source: Image.Image,
*,
upscale: UpscaleParams,
stage_source: Image.Image,
stage_source: Optional[Image.Image] = None,
**kwargs,
) -> Image.Image:
upscale = upscale.with_args(**kwargs)