(sampling_steps, shift)
| 22 | |
| 23 | |
| 24 | def get_sampling_sigmas(sampling_steps, shift): |
| 25 | sigma = np.linspace(1, 0, sampling_steps + 1)[:sampling_steps] |
| 26 | sigma = (shift * sigma / (1 + (shift - 1) * sigma)) |
| 27 | |
| 28 | return sigma |
| 29 | |
| 30 | |
| 31 | def retrieve_timesteps( |