ksampler

2024-03-21 22:28:14 -06:00
parent ef0686f8c7
commit f69dde2e62
1 changed files with 5 additions and 1 deletions
@@ -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, )