Fix regression.

Author: comfyanonymous

comfy_extras/nodes_model_advanced.py

@@ -26,8 +26,8 @@ def calculate_denoised(self, sigma, model_output, model_input):
 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