From 32165b27af13df03ed4b168bb3c7ea9fc2c5171c Mon Sep 17 00:00:00 2001 From: Haoming Date: Tue, 30 Dec 2025 10:07:52 +0800 Subject: [PATCH] taesd --- modules/sd_samplers_common.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/modules/sd_samplers_common.py b/modules/sd_samplers_common.py index 79f41a85..de2c27b2 100644 --- a/modules/sd_samplers_common.py +++ b/modules/sd_samplers_common.py @@ -94,13 +94,16 @@ def samples_to_image_grid(samples, approximation=None): def images_tensor_to_samples(image, approximation=None, model=None): """image[0, 1] -> latent""" + x_latent = None + if approximation is None: approximation = approximation_indexes.get(opts.sd_vae_encode_method, 0) if approximation == 3: - image = image.to(devices.device, devices.dtype) - x_latent = sd_vae_taesd.encoder_model()(image) - else: + if (mdl := sd_vae_taesd.encoder_model()) is not None: + x_latent = mdl(image.to(devices.device, devices.dtype)).detach() + + if x_latent is None: if model is None: model = shared.sd_model