fix: TAE encoding (#935)

This commit is contained in:
stduhpf 2025-11-07 15:58:59 +01:00 committed by GitHub
parent 8f6c5c217b
commit fb748bb8a4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1645,7 +1645,9 @@ public:
} else {
latent = gaussian_latent_sample(work_ctx, vae_output);
}
process_latent_in(latent);
if (!use_tiny_autoencoder) {
process_latent_in(latent);
}
if (sd_version_is_qwen_image(version)) {
latent = ggml_reshape_4d(work_ctx, latent, latent->ne[0], latent->ne[1], latent->ne[3], 1);
}