Forgot this.

jn-jairo · Dec 15, 2023 · adc40e3 · adc40e3
1 parent 014c8bf
commit adc40e3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/comfy_extras/nodes_model_advanced.py b/comfy_extras/nodes_model_advanced.py
@@ -92,7 +92,7 @@ def patch(self, model, sampling, zsnr):
         class ModelSamplingAdvanced(sampling_base, sampling_type):
             pass
 
-        model_sampling = ModelSamplingAdvanced()
+        model_sampling = ModelSamplingAdvanced(model.model.model_config)
         if zsnr:
             model_sampling.set_sigmas(rescale_zero_terminal_snr_sigmas(model_sampling.sigmas))