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
|
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
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -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,
|
||||||
|
|
Loading…
Reference in New Issue