fix: skip reference image VAE encoding when latents have no consumer (#2207)

This commit is contained in:
Wagner Bruna authored and GitHub committed 2026-10-25 05:19:32 +08:00
1 parent ea209dbfcf
commit 0886f11b6b
1 file changed
+5 -1
+5 -1
View File
@@ -313,9 +313,13 @@ namespace sd::pipeline {
LOG_INFO("EDIT mode"); LOG_INFO("EDIT mode");
} }
bool skip_vae_encoding = (sd->version == VERSION_HIDREAM_O1) || !(
ref_image_params.pass_to_dit || ref_image_params.vlm_use_vae_input ||
sd_version_is_unet_edit(sd->version) || sd_version_is_pid(sd->version));
std::vector<sd::Tensor<float>> ref_latents; std::vector<sd::Tensor<float>> ref_latents;
for (size_t i = 0; i < ref_images.size(); i++) { for (size_t i = 0; i < ref_images.size(); i++) {
if (sd->version == VERSION_HIDREAM_O1) { if (skip_vae_encoding) {
continue; continue;
} }
sd::Tensor<float> ref_latent; sd::Tensor<float> ref_latent;