Update smZ_cfg_denoiser.py

This commit is contained in:
VALADI K JAGANATHAN
2023-11-01 13:51:41 +05:30
committed by GitHub
parent 25a7bb2539
commit 5ce8cb49e8

View File

@@ -4,7 +4,7 @@ import torch
from typing import List
import comfy.sample
from comfy import model_base, model_management
from comfy.samplers import KSampler, CompVisVDenoiser, KSamplerX0Inpaint
from comfy.samplers import KSampler, KSamplerX0Inpaint
from comfy.k_diffusion.external import CompVisDenoiser
import nodes
import inspect
@@ -318,4 +318,4 @@ def set_model_k(self: KSampler):
class SDKSampler(comfy.samplers.KSampler):
def __init__(self, *args, **kwargs):
super(SDKSampler, self).__init__(*args, **kwargs)
set_model_k(self)
set_model_k(self)