Unverified Commit 1ce4b5f3 authored by Jianqi Pan's avatar Jianqi Pan Committed by GitHub
Browse files

fix: fix forward function signature of controlnet reference_only pipeline example (#5717)



fix: ignore other args
Co-authored-by: default avatarPatrick von Platen <patrick.v.platen@gmail.com>
parent c9f847a7
...@@ -546,7 +546,7 @@ class StableDiffusionControlNetReferencePipeline(StableDiffusionControlNetPipeli ...@@ -546,7 +546,7 @@ class StableDiffusionControlNetReferencePipeline(StableDiffusionControlNetPipeli
return hidden_states, output_states return hidden_states, output_states
def hacked_DownBlock2D_forward(self, hidden_states, temb=None): def hacked_DownBlock2D_forward(self, hidden_states, temb=None, *args, **kwargs):
eps = 1e-6 eps = 1e-6
output_states = () output_states = ()
...@@ -642,7 +642,7 @@ class StableDiffusionControlNetReferencePipeline(StableDiffusionControlNetPipeli ...@@ -642,7 +642,7 @@ class StableDiffusionControlNetReferencePipeline(StableDiffusionControlNetPipeli
return hidden_states return hidden_states
def hacked_UpBlock2D_forward(self, hidden_states, res_hidden_states_tuple, temb=None, upsample_size=None): def hacked_UpBlock2D_forward(self, hidden_states, res_hidden_states_tuple, temb=None, upsample_size=None, *args, **kwargs):
eps = 1e-6 eps = 1e-6
for i, resnet in enumerate(self.resnets): for i, resnet in enumerate(self.resnets):
# pop res hidden states # pop res hidden states
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment