Forgot this.
This commit is contained in:
parent
014c8bf2f2
commit
adc40e3d7b
|
@ -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))
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue