fix(api): add out channels to upscaling config

ssube · Nov 19, 2023 · e64e514 · e64e514
1 parent c8dd85e
commit e64e514
Showing 1 changed file with 6 additions and 0 deletions.
diff --git 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