Skip to content

GitLab

Commit 41b07ff8 authored by Jedrzej Kosinski's avatar Jedrzej Kosinski
Browse files

Fix TAESD preview to only decode first latent, instead of all

parent a12cc053
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
latent_preview.py
View file @ 41b07ff8
...@@ -22,7 +22,7 @@ class TAESDPreviewerImpl(LatentPreviewer): ...@@ -22,7 +22,7 @@ class TAESDPreviewerImpl(LatentPreviewer):
self.taesd = taesd self.taesd = taesd
def decode_latent_to_preview(self, x0): def decode_latent_to_preview(self, x0):
x_sample = self.taesd.decoder(x0)[0].detach() x_sample = self.taesd.decoder(x0[:1])[0].detach()
# x_sample = self.taesd.unscale_latents(x_sample).div(4).add(0.5) # returns value in [-2, 2] # x_sample = self.taesd.unscale_latents(x_sample).div(4).add(0.5) # returns value in [-2, 2]
x_sample = x_sample.sub(0.5).mul(2) x_sample = x_sample.sub(0.5).mul(2)
......
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