From be9f51b25cea5e6107b73adfc0f00282f1b12356 Mon Sep 17 00:00:00 2001 From: Wagner Bruna Date: Sat, 11 Apr 2026 06:18:23 -0300 Subject: [PATCH] refactor: simplify DiscreteFlowDenoiser (#1405) --- src/denoiser.hpp | 14 ++------------ 1 file changed, 2 insertions(+), 12 deletions(-) diff --git a/src/denoiser.hpp b/src/denoiser.hpp index a7292b93..c9c9d881 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 {