From 868b55b32557276b76d8d62fb00369a791db67f5 Mon Sep 17 00:00:00 2001 From: Emmanuel Benazera Date: Thu, 5 Oct 2023 19:26:38 +0000 Subject: [PATCH] fix: wrong number of model forward signature parameters check --- models/modules/palette_denoise_fn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/modules/palette_denoise_fn.py b/models/modules/palette_denoise_fn.py index 61dedec3b..b06354261 100644 --- a/models/modules/palette_denoise_fn.py +++ b/models/modules/palette_denoise_fn.py @@ -105,7 +105,7 @@ def forward(self, input, embed_noise_level, cls, mask, ref): if "mask" in self.conditioning: input = torch.cat([input, mask_embed], dim=1) - if self.model_nargs == 4: # ref from dataloader with reference image + if self.model_nargs == 3: # ref from dataloader with reference image out = self.model(input, embedding, ref) else: out = self.model(input, embedding)