From 0881c95aa8917da30949b23d8f96ef9bce7cd7cf Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?St=C3=A9phane=20du=20Hamel?= <stephduh@live.fr>
Date: Fri, 22 May 2026 17:32:04 +0200
Subject: [PATCH] fix --taesd-preview-only

---
 src/stable-diffusion.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/stable-diffusion.cpp b/src/stable-diffusion.cpp
index eb6845b46..36c44a75b 100644
--- a/src/stable-diffusion.cpp
+++ b/src/stable-diffusion.cpp
@@ -759,7 +759,7 @@ class StableDiffusionGGML {
                     LOG_INFO("using TAE for preview");
                     preview_vae = create_tae();
                     preview_vae->set_max_graph_vram_bytes(max_graph_vram_bytes);
-                    get_param_tensors_p(first_stage_model, vae_mmap, "vae");
+                    get_param_tensors_p(preview_vae, vae_mmap, "tae");
                 }
             }