DarthAffe/stable-diffusion.cpp
1
0
Fork 0
mirror of https://github.com/leejet/stable-diffusion.cpp.git synced 2025-12-12 21:38:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: simplify DPM++ (2S) Ancestral (#667)

Browse Source
This commit is contained in:
vmobilis 2025-09-16 18:05:25 +03:00 committed by GitHub
parent 8909523e92
commit 97ad3e7ff9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
denoiser.hpp
View File

@ -770,7 +770,6 @@ static void sample_k_diffusion(sample_method_t method,
} break; } break;
case DPMPP2S_A: { case DPMPP2S_A: {
struct ggml_tensor* noise = ggml_dup_tensor(work_ctx, x); struct ggml_tensor* noise = ggml_dup_tensor(work_ctx, x);
struct ggml_tensor* d = ggml_dup_tensor(work_ctx, x);
struct ggml_tensor* x2 = ggml_dup_tensor(work_ctx, x); struct ggml_tensor* x2 = ggml_dup_tensor(work_ctx, x);
for (int i = 0; i < steps; i++) { for (int i = 0; i < steps; i++) {
@ -785,22 +784,15 @@ static void sample_k_diffusion(sample_method_t method,
auto sigma_fn = [](float t) -> float { return exp(-t); }; auto sigma_fn = [](float t) -> float { return exp(-t); };
if (sigma_down == 0) { if (sigma_down == 0) {
// Euler step // d = (x - denoised) / sigmas[i];
float* vec_d = (float*)d->data; // dt = sigma_down - sigmas[i];
// x += d * dt;
// => x = denoised
float* vec_x = (float*)x->data; float* vec_x = (float*)x->data;
float* vec_denoised = (float*)denoised->data; float* vec_denoised = (float*)denoised->data;
for (int j = 0; j < ggml_nelements(d); j++) { for (int j = 0; j < ggml_nelements(x); j++) {
vec_d[j] = (vec_x[j] - vec_denoised[j]) / sigmas[i]; vec_x[j] = vec_denoised[j];
}
// TODO: If sigma_down == 0, isn't this wrong?
// But
// https://github.com/crowsonkb/k-diffusion/blob/master/k_diffusion/sampling.py#L525
// has this exactly the same way.
float dt = sigma_down - sigmas[i];
for (int j = 0; j < ggml_nelements(d); j++) {
vec_x[j] = vec_x[j] + vec_d[j] * dt;
} }
} else { } else {
// DPM-Solver++(2S) // DPM-Solver++(2S)
@ -809,7 +801,6 @@ static void sample_k_diffusion(sample_method_t method,
float h = t_next - t; float h = t_next - t;
float s = t + 0.5f * h; float s = t + 0.5f * h;
float* vec_d = (float*)d->data;
float* vec_x = (float*)x->data; float* vec_x = (float*)x->data;
float* vec_x2 = (float*)x2->data; float* vec_x2 = (float*)x2->data;
float* vec_denoised = (float*)denoised->data; float* vec_denoised = (float*)denoised->data;