apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): add current params to outpaint metadata

This commit is contained in:
Sean Sube 2024-03-09 12:30:07 -06:00
parent 08396c8847
commit ce48ce2d10
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -19,7 +19,7 @@ from ..server import ServerContext
from ..utils import is_debug
from ..worker import ProgressCallback, WorkerContext
from .base import BaseStage
from .result import StageResult
from .result import ImageMetadata, StageResult
logger = getLogger(__name__)
@ -123,4 +123,11 @@ class UpscaleOutpaintStage(BaseStage):
outputs.extend(result.images)
return StageResult(images=outputs, metadata=sources.metadata)
metadata = [
ImageMetadata(
params, size, inversions=inversions, loras=loras, ancestors=[source]
)
for source in sources.metadata
]
return StageResult(images=outputs, metadata=metadata)