DarthAffe/stable-diffusion.cpp
1
0
Fork 0
mirror of https://github.com/leejet/stable-diffusion.cpp.git synced 2026-05-08 16:28:53 +00: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 { struct DiscreteFlowDenoiser : public Denoiser {
float sigmas[TIMESTEPS];
float shift = 3.0f; float shift = 3.0f;
float sigma_data = 1.0f;
DiscreteFlowDenoiser(float shift = 3.0f) { DiscreteFlowDenoiser(float shift = 3.0f) {
set_shift(shift); 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) { void set_shift(float shift) {
this->shift = shift; this->shift = shift;
set_parameters();
} }
float sigma_min() override { float sigma_min() override {
return sigmas[0]; return t_to_sigma(0);
} }
float sigma_max() override { float sigma_max() override {
return sigmas[TIMESTEPS - 1]; return t_to_sigma(TIMESTEPS - 1);
} }
float sigma_to_t(float sigma) override { float sigma_to_t(float sigma) override {