diff --git a/examples/community/stable_diffusion_controlnet_reference.py b/examples/community/stable_diffusion_controlnet_reference.py index b18f276640..358fc1c6dc 100644 --- a/examples/community/stable_diffusion_controlnet_reference.py +++ b/examples/community/stable_diffusion_controlnet_reference.py @@ -642,7 +642,9 @@ class StableDiffusionControlNetReferencePipeline(StableDiffusionControlNetPipeli return hidden_states - def hacked_UpBlock2D_forward(self, hidden_states, res_hidden_states_tuple, temb=None, upsample_size=None, *args, **kwargs): + def hacked_UpBlock2D_forward( + self, hidden_states, res_hidden_states_tuple, temb=None, upsample_size=None, *args, **kwargs + ): eps = 1e-6 for i, resnet in enumerate(self.resnets): # pop res hidden states