Skip to content

Commit be9f51b

Browse files
authored
refactor: simplify DiscreteFlowDenoiser (#1405)
1 parent e8323ca commit be9f51b

File tree

1 file changed

+2
-12
lines changed

1 file changed

+2
-12
lines changed

src/denoiser.hpp

Lines changed: 2 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -658,32 +658,22 @@ inline float time_snr_shift(float alpha, float t) {
658658
}
659659

660660
struct DiscreteFlowDenoiser : public Denoiser {
661-
float sigmas[TIMESTEPS];
662661
float shift = 3.0f;
663662

664-
float sigma_data = 1.0f;
665-
666663
DiscreteFlowDenoiser(float shift = 3.0f) {
667664
set_shift(shift);
668665
}
669666

670-
void set_parameters() {
671-
for (int i = 0; i < TIMESTEPS; i++) {
672-
sigmas[i] = t_to_sigma(static_cast<float>(i));
673-
}
674-
}
675-
676667
void set_shift(float shift) {
677668
this->shift = shift;
678-
set_parameters();
679669
}
680670

681671
float sigma_min() override {
682-
return sigmas[0];
672+
return t_to_sigma(0);
683673
}
684674

685675
float sigma_max() override {
686-
return sigmas[TIMESTEPS - 1];
676+
return t_to_sigma(TIMESTEPS - 1);
687677
}
688678

689679
float sigma_to_t(float sigma) override {

0 commit comments

Comments
 (0)