Skip to content

Commit a3a2b2d

Browse files
committed
fix get_first_stage_encoding
1 parent a8d3aa0 commit a3a2b2d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

stable-diffusion.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1510,7 +1510,7 @@ class StableDiffusionGGML {
15101510

15111511
ggml_tensor* get_first_stage_encoding(ggml_context* work_ctx, ggml_tensor* vae_output) {
15121512
ggml_tensor* latent;
1513-
if (use_tiny_autoencoder || sd_version_is_qwen_image(version)) {
1513+
if (use_tiny_autoencoder || sd_version_is_qwen_image(version) || sd_version_is_wan(version)) {
15141514
latent = vae_output;
15151515
} else if (version == VERSION_SD1_PIX2PIX) {
15161516
latent = ggml_view_3d(work_ctx,

0 commit comments

Comments
 (0)