apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix outpaint size calculation

This commit is contained in:
Sean Sube 2024-03-09 12:52:04 -06:00
parent f785548fb8
commit 322aa3fd7f
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -61,6 +61,7 @@ class UpscaleOutpaintStage(BaseStage):
loras=loras,
)
sizes = []
outputs = []
for source in sources.as_images():
if is_debug():
@ -121,13 +122,14 @@ class UpscaleOutpaintStage(BaseStage):
callback=callback,
)
sizes.extend([size] * len(result.images))
outputs.extend(result.images)
metadata = [
ImageMetadata(
params, size, inversions=inversions, loras=loras, ancestors=[source]
)
for source in sources.metadata
for size, source in zip(sizes, sources.metadata)
]
return StageResult(images=outputs, metadata=metadata)