From 858bfc7a3266e919bb8783c496dda943e58aed31 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Sun, 4 Feb 2024 19:40:21 -0800 Subject: [PATCH] Update unet_patcher.py --- 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 a20ccb70..4c00fa84 100644 --- a/modules_forge/unet_patcher.py +++ b/modules_forge/unet_patcher.py @@ -102,7 +102,7 @@ class UnetPatcher(ModelPatcher): self.append_model_option('conditioning_modifiers', modifier, ensure_uniqueness) return - def add_memory_peak_estimation_modifier(self, modifier): + def set_memory_peak_estimation_modifier(self, modifier): self.model_options['memory_peak_estimation_modifier'] = modifier return