From 8c9dc0d0dc009b71f4e172a1889e00e40e2fceea Mon Sep 17 00:00:00 2001 From: Chengsong Zhang Date: Sat, 3 Feb 2024 21:22:20 -0600 Subject: [PATCH] retutn patcher to calc model size (#34) --- modules_forge/unet_patcher.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules_forge/unet_patcher.py b/modules_forge/unet_patcher.py index ad52b093..e6992206 100644 --- a/modules_forge/unet_patcher.py +++ b/modules_forge/unet_patcher.py @@ -58,7 +58,7 @@ class UnetPatcher(ModelPatcher): patcher = ModelPatcher(model=m, load_device=self.load_device, offload_device=self.offload_device) self.add_extra_model_patcher_during_sampling(patcher) - return + return patcher def add_patched_controlnet(self, cnet): cnet.set_previous_controlnet(self.controlnet_linked_list)