modelzoo.vision.pytorch.dit.samplers.sampler_utils.rescale_zero_terminal_snr

modelzoo.vision.pytorch.dit.samplers.sampler_utils.rescale_zero_terminal_snr(betas)

Rescales betas to have zero terminal SNR Based on https://arxiv.org/pdf/2305.08891.pdf (Algorithm 1)

Parameters

betas (torch.FloatTensor) – the betas that the scheduler is being initialized with.

Returns

rescaled betas with zero terminal SNR

Return type

torch.FloatTensor

previous

modelzoo.vision.pytorch.dit.samplers.sampler_utils

next

modelzoo.vision.pytorch.dit.samplers.sampler_utils.set_sampling_timesteps