diff --git a/api/onnx_web/convert/diffusion/checkpoint.py b/api/onnx_web/convert/diffusion/checkpoint.py index 649891620..c9b4f1034 100644 --- a/api/onnx_web/convert/diffusion/checkpoint.py +++ b/api/onnx_web/convert/diffusion/checkpoint.py @@ -1539,7 +1539,7 @@ def extract_checkpoint( ) unet_config["upcast_attention"] = upcast_attention if is_inpainting: - unet_config["in_channels "] = 9 + unet_config["in_channels"] = 9 unet = UNet2DConditionModel(**unet_config)