alphard/efficiency-nodes-comfyui
1
0
Fork 0
mirror of https://github.com/jags111/efficiency-nodes-comfyui.git synced 2026-03-21 21:22:13 -03:00
Code Issues Packages Projects Releases Wiki Activity

added fix for smZ_cfg_denoiser by zdouble https://github.com/jags111/efficiency-nodes-comfyui/pull/12

Browse Source
This commit is contained in:
dnl13
2023-11-03 12:16:13 +01:00
parent 06fead1e87
commit d00c1031a2
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
py/smZ_cfg_denoiser.py
View File

@@ -4,7 +4,7 @@ import torch
from typing import List from typing import List
import comfy.sample import comfy.sample
from comfy import model_base, model_management from comfy import model_base, model_management
from comfy.samplers import KSampler, KSamplerX0Inpaint from comfy.samplers import KSampler, KSamplerX0Inpaint, wrap_model
#from comfy.k_diffusion.external import CompVisDenoiser #from comfy.k_diffusion.external import CompVisDenoiser
import nodes import nodes
import inspect import inspect
@@ -308,10 +308,10 @@ def set_model_k(self: KSampler):
self.model_denoise = CFGNoisePredictor(self.model) # main change self.model_denoise = CFGNoisePredictor(self.model) # main change
if ((getattr(self.model, "parameterization", "") == "v") or if ((getattr(self.model, "parameterization", "") == "v") or
(getattr(self.model, "model_type", -1) == model_base.ModelType.V_PREDICTION)): (getattr(self.model, "model_type", -1) == model_base.ModelType.V_PREDICTION)):
self.model_wrap = CompVisVDenoiser(self.model_denoise, quantize=True) self.model_wrap = wrap_model(self.model_denoise, quantize=True)
self.model_wrap.parameterization = getattr(self.model, "parameterization", "v") self.model_wrap.parameterization = getattr(self.model, "parameterization", "v")
else: else:
self.model_wrap = CompVisDenoiser(self.model_denoise, quantize=True) self.model_wrap = wrap_model(self.model_denoise, quantize=True)
self.model_wrap.parameterization = getattr(self.model, "parameterization", "eps") self.model_wrap.parameterization = getattr(self.model, "parameterization", "eps")
self.model_k = KSamplerX0Inpaint(self.model_wrap) self.model_k = KSamplerX0Inpaint(self.model_wrap)