Skip to content

Commit fb748bb

Browse files
authored
fix: TAE encoding (#935)
1 parent 8f6c5c2 commit fb748bb

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

stable-diffusion.cpp

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1645,7 +1645,9 @@ class StableDiffusionGGML {
16451645
} else {
16461646
latent = gaussian_latent_sample(work_ctx, vae_output);
16471647
}
1648-
process_latent_in(latent);
1648+
if (!use_tiny_autoencoder) {
1649+
process_latent_in(latent);
1650+
}
16491651
if (sd_version_is_qwen_image(version)) {
16501652
latent = ggml_reshape_4d(work_ctx, latent, latent->ne[0], latent->ne[1], latent->ne[3], 1);
16511653
}

0 commit comments

Comments
 (0)