Remove wrong scale

This commit is contained in:
Kohaku-Blueleaf 2023-08-16 11:21:12 +08:00
parent 85fcb7b8df
commit d9ddc5d4cd
1 changed files with 1 additions and 2 deletions

View File

@ -44,8 +44,7 @@ def samples_to_images_tensor(sample, approximation=None, model=None):
elif approximation == 1: elif approximation == 1:
x_sample = sd_vae_approx.model()(sample.to(devices.device, devices.dtype)).detach() x_sample = sd_vae_approx.model()(sample.to(devices.device, devices.dtype)).detach()
elif approximation == 3: elif approximation == 3:
x_sample = sample * 1.5 x_sample = sd_vae_taesd.decoder_model()(sample.to(devices.device, devices.dtype)).detach()
x_sample = sd_vae_taesd.decoder_model()(x_sample.to(devices.device, devices.dtype)).detach()
x_sample = x_sample * 2 - 1 x_sample = x_sample * 2 - 1
else: else:
if model is None: if model is None: