Fix VAE preview darkening

leejet · Dec 13, 2024 · 48fcec2 · 48fcec2
1 parent 86450ce
commit 48fcec2
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/stable-diffusion.cpp b/stable-diffusion.cpp
@@ -839,6 +839,7 @@ class StableDiffusionGGML {
                     first_stage_model->compute(n_threads, latents, true, &result);
                 }
                 first_stage_model->free_compute_buffer();
+                ggml_tensor_scale(latents, scale_factor);
 
                 ggml_tensor_scale_output(result);
             } else if (preview_mode == SD_PREVIEW_TAE) {