Forgot this.

This commit is contained in:
comfyanonymous 2023-12-15 15:46:23 -05:00
parent 014c8bf2f2
commit adc40e3d7b
1 changed files with 1 additions and 1 deletions

View File

@ -92,7 +92,7 @@ class ModelSamplingDiscrete:
class ModelSamplingAdvanced(sampling_base, sampling_type): class ModelSamplingAdvanced(sampling_base, sampling_type):
pass pass
model_sampling = ModelSamplingAdvanced() model_sampling = ModelSamplingAdvanced(model.model.model_config)
if zsnr: if zsnr:
model_sampling.set_sigmas(rescale_zero_terminal_snr_sigmas(model_sampling.sigmas)) model_sampling.set_sigmas(rescale_zero_terminal_snr_sigmas(model_sampling.sigmas))