From e2bc5e54b5257681858e85d8b7a7ad10c8024a78 Mon Sep 17 00:00:00 2001 From: patil-suraj Date: Fri, 2 Dec 2022 14:30:47 +0100 Subject: [PATCH] fix decodeing --- src/diffusers/models/vae.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/models/vae.py b/src/diffusers/models/vae.py index ab19422156..ab8f904e52 100644 --- a/src/diffusers/models/vae.py +++ b/src/diffusers/models/vae.py @@ -749,7 +749,7 @@ class AutoencoderKL(ModelMixin, ConfigMixin): # 2. apply model loop over last dim - output_list = [self._decode(z).sample(z[:, :, :, :, i]) for i in range(z.shape[-1])] + output_list = [self._decode(z[:, :, :, :, i]).sample for i in range(z.shape[-1])] o = torch.stack(output_list, axis=-1) # # (bn, nc, ks[0], ks[1], L) o = o * weighting