From 8cf55a3b3b408268bff94f1727f755ae1f3bcde4 Mon Sep 17 00:00:00 2001
From: stduhpf <stephduh@live.fr>
Date: Fri, 22 May 2026 18:22:35 +0200
Subject: [PATCH] fix: load TAESD preview-only model correctly (#1547)

---
 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 eb6845b4..36c44a75 100644
--- a/src/stable-diffusion.cpp
+++ b/src/stable-diffusion.cpp
@@ -759,7 +759,7 @@ public:
                     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");
                 }
             }