From f69dde2e62e292952722cec2765ed745dac0cdef Mon Sep 17 00:00:00 2001
From: logtd <logoftd@gmail.com>
Date: Thu, 21 Mar 2024 22:28:14 -0600
Subject: [PATCH] ksampler

---
 nodes.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/nodes.py b/nodes.py
index d774b8a..80b90e8 100644
--- a/nodes.py
+++ b/nodes.py
@@ -1,6 +1,8 @@
 import torch
 from tqdm import trange
 
+from comfy.samplers import KSAMPLER
+
 
 class SamplerInversedEulerNode:
     @classmethod
@@ -39,4 +41,6 @@ class SamplerInversedEulerNode:
                     callback(
                         {'x': x, 'i': i, 'sigma': sigmas[i], 'sigma_hat': sigmas[i], 'denoised': denoised})
             return x
-        return (sample_inversed_euler, )
+
+        ksampler = KSAMPLER(sample_inversed_euler)
+        return (ksampler, )