From a2a2b6023fed815ecacdba315130187780269d96 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Thu, 25 Jan 2024 16:24:35 -0800 Subject: [PATCH] Update forge_loader.py --- modules_forge/forge_loader.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modules_forge/forge_loader.py b/modules_forge/forge_loader.py index 4fa6c977..33cb7fde 100644 --- a/modules_forge/forge_loader.py +++ b/modules_forge/forge_loader.py @@ -243,8 +243,8 @@ def load_model_for_a1111(timer, checkpoint_info=None, state_dict=None): original_forward = sd_model.model.diffusion_model.forward - def forge_unet_forward(self, *args, **kwargs): - return original_forward(self, *args, **kwargs) + def forge_unet_forward(*args, **kwargs): + return original_forward(*args, **kwargs) sd_model.model.diffusion_model.forward = forge_unet_forward