diff --git a/api/onnx_web/diffusion/run.py b/api/onnx_web/diffusion/run.py index 8cac1c65..f1436c96 100644 --- a/api/onnx_web/diffusion/run.py +++ b/api/onnx_web/diffusion/run.py @@ -253,6 +253,7 @@ def run_blend_pipeline( mask=mask, callback=progress, ) + image = image.convert("RGB") image = run_upscale_correction( job, server, stage, params, image, upscale=upscale, callback=progress