From 84b15843a82bacb124322ee377244b152e407219 Mon Sep 17 00:00:00 2001 From: Wagner Bruna Date: Fri, 10 Apr 2026 00:44:05 -0300 Subject: [PATCH] refactor: simplify DiscreteFlowDenoiser sigma_data isn't used anywhere. The sigmas array is only used for sigma_min and sigma_max, so replace it with a direct calculation. --- src/denoiser.hpp | 14 ++------------ 1 file changed, 2 insertions(+), 12 deletions(-) diff --git a/src/denoiser.hpp b/src/denoiser.hpp index a7292b93d..c9c9d881d 100644 --- a/src/denoiser.hpp +++ b/src/denoiser.hpp @@ -658,32 +658,22 @@ inline float time_snr_shift(float alpha, float t) { } struct DiscreteFlowDenoiser : public Denoiser { - float sigmas[TIMESTEPS]; float shift = 3.0f; - float sigma_data = 1.0f; - DiscreteFlowDenoiser(float shift = 3.0f) { set_shift(shift); } - void set_parameters() { - for (int i = 0; i < TIMESTEPS; i++) { - sigmas[i] = t_to_sigma(static_cast(i)); - } - } - void set_shift(float shift) { this->shift = shift; - set_parameters(); } float sigma_min() override { - return sigmas[0]; + return t_to_sigma(0); } float sigma_max() override { - return sigmas[TIMESTEPS - 1]; + return t_to_sigma(TIMESTEPS - 1); } float sigma_to_t(float sigma) override {