From 75f1a2ceade1fd604aff277d77738cb6e58ab27b Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Sun, 26 Nov 2023 10:43:08 -0600
Subject: [PATCH] pass results object to chain pipeline from endpoint

---
 api/onnx_web/diffusers/pipelines/upscale.py | 6 +-----
 api/onnx_web/server/api.py                  | 3 ++-
 2 files changed, 3 insertions(+), 6 deletions(-)

diff --git a/api/onnx_web/diffusers/pipelines/upscale.py b/api/onnx_web/diffusers/pipelines/upscale.py
index fc0e477b..e9df3a5f 100644
--- 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
index bbd5f038..a9162f0f 100644
--- a/api/onnx_web/server/api.py
+++ b/api/onnx_web/server/api.py
@@ -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,