youngkingdom/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-06 02:07:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add a way for nodes to set a custom CFG function.

Browse Source
This commit is contained in:
comfyanonymous
2023-04-17 11:05:15 -04:00
parent 0ab5c619ea
commit 884ea653c8
2 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
comfy/sd.py
View File

@ -250,6 +250,9 @@ class ModelPatcher:
def set_model_tomesd(self, ratio):
self.model_options["transformer_options"]["tomesd"] = {"ratio": ratio}
def set_model_sampler_cfg_function(self, sampler_cfg_function):
self.model_options["sampler_cfg_function"] = sampler_cfg_function
def model_dtype(self):
return self.model.diffusion_model.dtype