From e64e514e7fb55cf1c3e7a1285ea7b505d6675253 Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Sat, 18 Nov 2023 21:35:57 -0600
Subject: [PATCH] fix(api): add out channels to upscaling config

---
 api/onnx_web/diffusers/pipelines/upscale.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/api/onnx_web/diffusers/pipelines/upscale.py b/api/onnx_web/diffusers/pipelines/upscale.py
index deb2c8c3..a9c6f843 100644
--- a/api/onnx_web/diffusers/pipelines/upscale.py
+++ b/api/onnx_web/diffusers/pipelines/upscale.py
@@ -51,9 +51,15 @@ def preprocess(image):
 
 
 class FakeConfig:
+    block_out_channels: List[int]
     scaling_factor: float
 
     def __init__(self) -> None:
+        self.block_out_channels = [
+            128,
+            256,
+            512
+        ]
         self.scaling_factor = 0.08333