diff --git a/python_coreml_stable_diffusion/pipeline.py b/python_coreml_stable_diffusion/pipeline.py index 32ec718..1f495fa 100644 --- a/python_coreml_stable_diffusion/pipeline.py +++ b/python_coreml_stable_diffusion/pipeline.py @@ -517,7 +517,10 @@ def __call__( control_net_additional_residuals = {} # predict the noise residual - unet_additional_kwargs = unet_additional_kwargs | control_net_additional_residuals + unet_additional_kwargs = { + **unet_additional_kwargs, + **control_net_additional_residuals, + } noise_pred = self.unet( sample=latent_model_input.astype(np.float16),