Commit d9ddc5d4 authored by Kohaku-Blueleaf's avatar Kohaku-Blueleaf

Remove wrong scale

parent 85fcb7b8
...@@ -44,8 +44,7 @@ def samples_to_images_tensor(sample, approximation=None, model=None): ...@@ -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:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment