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

This commit is contained in:
stduhpf 2026-05-22 18:22:35 +02:00 committed by GitHub
parent 3a8788cb7d
commit 8cf55a3b3b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -759,7 +759,7 @@ public:
LOG_INFO("using TAE for preview"); LOG_INFO("using TAE for preview");
preview_vae = create_tae(); preview_vae = create_tae();
preview_vae->set_max_graph_vram_bytes(max_graph_vram_bytes); 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");
} }
} }