From fb748bb8a48edf0ba0424d3ee761eb0af71ab0f5 Mon Sep 17 00:00:00 2001
From: stduhpf <stephduh@live.fr>
Date: Fri, 7 Nov 2025 15:58:59 +0100
Subject: [PATCH] fix: TAE encoding (#935)

---
 stable-diffusion.cpp | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/stable-diffusion.cpp b/stable-diffusion.cpp
index 9faba95..d285f6c 100644
--- a/stable-diffusion.cpp
+++ b/stable-diffusion.cpp
@@ -1645,7 +1645,9 @@ public:
         } else {
             latent = gaussian_latent_sample(work_ctx, vae_output);
         }
-        process_latent_in(latent);
+        if (!use_tiny_autoencoder) {
+            process_latent_in(latent);
+        }
         if (sd_version_is_qwen_image(version)) {
             latent = ggml_reshape_4d(work_ctx, latent, latent->ne[0], latent->ne[1], latent->ne[3], 1);
         }