Fix VAE preview darkening

stduhpf · stduhpf · commit 737b3dd66a7a · 2025-02-06T16:03:30.000+01:00
diff --git a/stable-diffusion.cpp b/stable-diffusion.cpp
@@ -854,6 +854,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) {

Original file line number	Diff line number	Diff line change
`@@ -854,6 +854,7 @@ class StableDiffusionGGML {`
`854`	`854`	`first_stage_model->compute(n_threads, latents, true, &result);`
`855`	`855`	`}`
`856`	`856`	`first_stage_model->free_compute_buffer();`
	`857`	`+ ggml_tensor_scale(latents, scale_factor);`
`857`	`858`
`858`	`859`	`ggml_tensor_scale_output(result);`
`859`	`860`	`} else if (preview_mode == SD_PREVIEW_TAE) {`