fix: load TAESD preview-only model correctly (#1547)

2026-10-01 02:18:01 -05:00 · 2026-05-23 00:22:35 +08:00
parent 3a8788cb7d
commit 8cf55a3b3b
1 changed files with 1 additions and 1 deletions
@@ -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");
                }
            }