apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

make progress callbacks optional

This commit is contained in:
Sean Sube 2023-02-12 12:22:11 -06:00
parent d9fc908592
commit 06d4f32574
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
4 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/onnx_web/chain/blend_img2img.py
View File

@ -23,7 +23,7 @@ def blend_img2img(
*, *,
strength: float, strength: float,
prompt: Optional[str] = None, prompt: Optional[str] = None,
callback: ProgressCallback, callback: ProgressCallback = None,
**kwargs, **kwargs,
) -> Image.Image: ) -> Image.Image:
prompt = prompt or params.prompt prompt = prompt or params.prompt

2
api/onnx_web/chain/blend_inpaint.py
View File

@ -29,7 +29,7 @@ def blend_inpaint(
fill_color: str = "white", fill_color: str = "white",
mask_filter: Callable = mask_filter_none, mask_filter: Callable = mask_filter_none,
noise_source: Callable = noise_source_histogram, noise_source: Callable = noise_source_histogram,
callback: ProgressCallback, callback: ProgressCallback = None,
**kwargs, **kwargs,
) -> Image.Image: ) -> Image.Image:
logger.info("upscaling image by expanding borders", expand) logger.info("upscaling image by expanding borders", expand)

2
api/onnx_web/chain/upscale_outpaint.py
View File

@ -30,7 +30,7 @@ def upscale_outpaint(
fill_color: str = "white", fill_color: str = "white",
mask_filter: Callable = mask_filter_none, mask_filter: Callable = mask_filter_none,
noise_source: Callable = noise_source_histogram, noise_source: Callable = noise_source_histogram,
callback: ProgressCallback, callback: ProgressCallback = None,
**kwargs, **kwargs,
) -> Image.Image: ) -> Image.Image:
prompt = prompt or params.prompt prompt = prompt or params.prompt

2
api/onnx_web/chain/upscale_stable_diffusion.py
View File

@ -67,7 +67,7 @@ def upscale_stable_diffusion(
*, *,
upscale: UpscaleParams, upscale: UpscaleParams,
prompt: str = None, prompt: str = None,
callback: ProgressCallback, callback: ProgressCallback = None,
**kwargs, **kwargs,
) -> Image.Image: ) -> Image.Image:
prompt = prompt or params.prompt prompt = prompt or params.prompt