Skip to content

Commit 5e16483

Browse files
stduhpfSkutteOleg
authored andcommitted
Fix TAE encoding
1 parent f1d01bd commit 5e16483

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
@@ -1726,7 +1726,9 @@ class StableDiffusionGGML {
17261726
} else {
17271727
latent = gaussian_latent_sample(work_ctx, vae_output);
17281728
}
1729-
process_latent_in(latent);
1729+
if (!use_tiny_autoencoder) {
1730+
process_latent_in(latent);
1731+
}
17301732
if (sd_version_is_qwen_image(version)) {
17311733
latent = ggml_reshape_4d(work_ctx, latent, latent->ne[0], latent->ne[1], latent->ne[3], 1);
17321734
}

0 commit comments

Comments
 (0)