Forgot this.

adc40e3d · comfyanonymous · 014c8bf2 · adc40e3d
Commit adc40e3d authored Dec 15, 2023 by comfyanonymous
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

comfy_extras/nodes_model_advanced.py comfy_extras/nodes_model_advanced.py +1 -1

No files found.
--- a/comfy_extras/nodes_model_advanced.py
+++ b/comfy_extras/nodes_model_advanced.py
@@ -92,7 +92,7 @@ class ModelSamplingDiscrete:
        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))