From 3748e7ef7a42197755f30a9b68f3e9b86d59e354 Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Wed, 13 Nov 2024 04:24:48 -0500 Subject: [PATCH] Fix regression. --- comfy_extras/nodes_model_advanced.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/comfy_extras/nodes_model_advanced.py b/comfy_extras/nodes_model_advanced.py index ed14b61a..f085bf12 100644 --- a/comfy_extras/nodes_model_advanced.py +++ b/comfy_extras/nodes_model_advanced.py @@ -26,8 +26,8 @@ class X0(comfy.model_sampling.EPS): class ModelSamplingDiscreteDistilled(comfy.model_sampling.ModelSamplingDiscrete): original_timesteps = 50 - def __init__(self, model_config=None): - super().__init__(model_config) + def __init__(self, model_config=None, zsnr=None): + super().__init__(model_config, zsnr=zsnr) self.skip_steps = self.num_timesteps // self.original_timesteps