apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

pass results object to chain pipeline from endpoint

This commit is contained in:
Sean Sube 2023-11-26 10:43:08 -06:00
parent 7ca5dc12a4
commit 75f1a2cead
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 3 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/onnx_web/diffusers/pipelines/upscale.py
View File

@ -55,11 +55,7 @@ class FakeConfig:
scaling_factor: float scaling_factor: float
def __init__(self) -> None: def __init__(self) -> None:
self.block_out_channels = [ self.block_out_channels = [128, 256, 512]
128,
256,
512
]
self.scaling_factor = 0.08333 self.scaling_factor = 0.08333

3
api/onnx_web/server/api.py
View File

@ -8,6 +8,7 @@ from jsonschema import validate
from PIL import Image from PIL import Image
from ..chain import CHAIN_STAGES, ChainPipeline from ..chain import CHAIN_STAGES, ChainPipeline
from ..chain.result import StageResult
from ..diffusers.load import get_available_pipelines, get_pipeline_schedulers from ..diffusers.load import get_available_pipelines, get_pipeline_schedulers
from ..diffusers.run import ( from ..diffusers.run import (
run_blend_pipeline, run_blend_pipeline,
@ -472,7 +473,7 @@ def chain(server: ServerContext, pool: DevicePoolExecutor):
pipeline, pipeline,
server, server,
base_params, base_params,
[], StageResult.empty(),
output=output, output=output,
size=base_size, size=base_size,
needs_device=device, needs_device=device,