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