pass results object to chain pipeline from endpoint

ssube · Nov 26, 2023 · 75f1a2c · 75f1a2c
1 parent 7ca5dc1
commit 75f1a2c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 6 deletions.
diff --git a/api/onnx_web/diffusers/pipelines/upscale.py b/api/onnx_web/diffusers/pipelines/upscale.py
@@ -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
 
 

diff --git a/api/onnx_web/server/api.py b/api/onnx_web/server/api.py
@@ -8,6 +8,7 @@
 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,