File tree Expand file tree Collapse file tree 1 file changed +2
-12
lines changed
Expand file tree Collapse file tree 1 file changed +2
-12
lines changed Original file line number Diff line number Diff line change @@ -658,32 +658,22 @@ inline float time_snr_shift(float alpha, float t) {
658658}
659659
660660struct 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 {
You can’t perform that action at this time.
0 commit comments