From 5ce8cb49e80d4860404bf503afae74bb18bd4592 Mon Sep 17 00:00:00 2001
From: VALADI K JAGANATHAN <info@orchidsasia.com>
Date: Wed, 1 Nov 2023 13:51:41 +0530
Subject: [PATCH] Update smZ_cfg_denoiser.py

---
 py/smZ_cfg_denoiser.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/py/smZ_cfg_denoiser.py b/py/smZ_cfg_denoiser.py
index e6a4ce1..54e61c6 100644
--- a/py/smZ_cfg_denoiser.py
+++ b/py/smZ_cfg_denoiser.py
@@ -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)
\ No newline at end of file
+        set_model_k(self)