CodeCow/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-07-21 21:01:24 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

taesd

Browse Source
This commit is contained in:
Haoming 2025-12-30 10:07:52 +08:00
parent b5c5d1176b
commit 32165b27af
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
modules/sd_samplers_common.py
View File

@ -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