diff --git a/src/denoiser.hpp b/src/denoiser.hpp index 59b8c41b..a7292b93 100644 --- a/src/denoiser.hpp +++ b/src/denoiser.hpp @@ -668,8 +668,8 @@ struct DiscreteFlowDenoiser : public Denoiser { } void set_parameters() { - for (int i = 1; i < TIMESTEPS + 1; i++) { - sigmas[i - 1] = t_to_sigma(static_cast(i)); + for (int i = 0; i < TIMESTEPS; i++) { + sigmas[i] = t_to_sigma(static_cast(i)); } }