pass results object to chain pipeline from endpoint
This commit is contained in:
parent
7ca5dc12a4
commit
75f1a2cead
|
@ -55,11 +55,7 @@ class FakeConfig:
|
|||
scaling_factor: float
|
||||
|
||||
def __init__(self) -> None:
|
||||
self.block_out_channels = [
|
||||
128,
|
||||
256,
|
||||
512
|
||||
]
|
||||
self.block_out_channels = [128, 256, 512]
|
||||
self.scaling_factor = 0.08333
|
||||
|
||||
|
||||
|
|
|
@ -8,6 +8,7 @@ from jsonschema import validate
|
|||
from PIL import Image
|
||||
|
||||
from ..chain import CHAIN_STAGES, ChainPipeline
|
||||
from ..chain.result import StageResult
|
||||
from ..diffusers.load import get_available_pipelines, get_pipeline_schedulers
|
||||
from ..diffusers.run import (
|
||||
run_blend_pipeline,
|
||||
|
@ -472,7 +473,7 @@ def chain(server: ServerContext, pool: DevicePoolExecutor):
|
|||
pipeline,
|
||||
server,
|
||||
base_params,
|
||||
[],
|
||||
StageResult.empty(),
|
||||
output=output,
|
||||
size=base_size,
|
||||
needs_device=device,
|
||||
|
|
Loading…
Reference in New Issue