git/stable-diffusion.cpp
1
0
Fork 0
mirror of https://github.com/leejet/stable-diffusion.cpp synced 2026-04-20 22:06:07 +02:00
Code Issues Packages Projects Releases Wiki Activity

refactor: simplify DiscreteFlowDenoiser (#1405)

Browse Source
This commit is contained in:
Wagner Bruna 2026-04-11 06:18:23 -03:00 committed by GitHub
parent e8323cabb0
commit be9f51b25c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/denoiser.hpp
View File

@ -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<float>(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 {