apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): match panorama img2img parameter check to numpy code path

This commit is contained in:
Sean Sube 2023-04-28 14:22:47 -05:00
parent 1d9de2c45b
commit c8e1519419
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -848,8 +848,8 @@ class OnnxStableDiffusionPanoramaPipeline(DiffusionPipeline):
*args,
**kwargs,
):
if len(args) > 0 and (
isinstance(args[0], np.ndarray) or isinstance(args[0], PIL.Image.Image)
if len(args) > 1 and (
isinstance(args[1], np.ndarray) or isinstance(args[1], PIL.Image.Image)
):
logger.debug("running img2img panorama pipeline")
return self.img2img(*args, **kwargs)