diff --git a/api/onnx_web/diffusers/pipelines/upscale.py b/api/onnx_web/diffusers/pipelines/upscale.py index ff571f83..fc0e477b 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