fix(api): remove trailing space from param name

ssube · Jun 10, 2023 · ac2eceb · ac2eceb
1 parent e64b813
commit ac2eceb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)