Fix VAE preview darkening

stduhpf · stduhpf · commit 48fcec2b482a · 2024-12-13T19:43:28.000+01:00
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) {

Original file line number	Diff line number	Diff line change
`@@ -839,6 +839,7 @@ class StableDiffusionGGML {`
`839`	`839`	`first_stage_model->compute(n_threads, latents, true, &result);`
`840`	`840`	`}`
`841`	`841`	`first_stage_model->free_compute_buffer();`
	`842`	`+ ggml_tensor_scale(latents, scale_factor);`
`842`	`843`
`843`	`844`	`ggml_tensor_scale_output(result);`
`844`	`845`	`} else if (preview_mode == SD_PREVIEW_TAE) {`