Update unet_patcher.py

This commit is contained in:
lllyasviel
2024-02-04 19:40:21 -08:00
parent fdbe5b7aa7
commit 858bfc7a32

View File

@@ -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