lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-27 18:23:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

add_sampler_pre_cfg_function

Browse Source
This commit is contained in:
lllyasviel
2024-02-05 14:48:43 -08:00
parent 8f86e66e5c
commit 88f6df4dcd
3 changed files with 13 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modules_forge/unet_patcher.py
View File

@@ -102,6 +102,10 @@ class UnetPatcher(ModelPatcher):
self.append_model_option('conditioning_modifiers', modifier, ensure_uniqueness)
return
def add_sampler_pre_cfg_function(self, modifier, ensure_uniqueness=False):
self.append_model_option('sampler_pre_cfg_function', modifier, ensure_uniqueness)
return
def set_memory_peak_estimation_modifier(self, modifier):
self.model_options['memory_peak_estimation_modifier'] = modifier
return