apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): make panorama switch mode based on kwargs

This commit is contained in:
Sean Sube 2023-07-01 15:04:12 -05:00
parent db86108dc5
commit 4b84188f6e
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
api/onnx_web/diffusers/pipelines/panorama.py
View File

@ -1243,11 +1243,18 @@ class OnnxStableDiffusionPanoramaPipeline(DiffusionPipeline):
*args,
**kwargs,
):
if len(args) > 1 and (
isinstance(args[1], np.ndarray) or isinstance(args[1], PIL.Image.Image)
):
if len(args) > 2 and (
if "image" in kwargs or (
len(args) > 1
and (
isinstance(args[1], np.ndarray) or isinstance(args[1], PIL.Image.Image)
)
):
if "mask_image" in kwargs or (
len(args) > 2
and (
isinstance(args[1], np.ndarray)
or isinstance(args[1], PIL.Image.Image)
)
):
logger.debug("running inpaint panorama pipeline")
return self.inpaint(*args, **kwargs)