From a3a2b2d72192dcd8a27cfa1b5d4ea53492596228 Mon Sep 17 00:00:00 2001 From: leejet Date: Thu, 25 Sep 2025 00:41:56 +0800 Subject: [PATCH] fix get_first_stage_encoding --- stable-diffusion.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/stable-diffusion.cpp b/stable-diffusion.cpp index db06508..60966ae 100644 --- a/stable-diffusion.cpp +++ b/stable-diffusion.cpp @@ -1510,7 +1510,7 @@ public: ggml_tensor* get_first_stage_encoding(ggml_context* work_ctx, ggml_tensor* vae_output) { ggml_tensor* latent; - if (use_tiny_autoencoder || sd_version_is_qwen_image(version)) { + if (use_tiny_autoencoder || sd_version_is_qwen_image(version) || sd_version_is_wan(version)) { latent = vae_output; } else if (version == VERSION_SD1_PIX2PIX) { latent = ggml_view_3d(work_ctx,