From 976173a4bfab7ba5a091a1901e7ebb9d69d2e32d Mon Sep 17 00:00:00 2001 From: patil-suraj Date: Wed, 29 Jun 2022 12:34:28 +0200 Subject: [PATCH] style --- 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 1a4c8904e4..bfc4a96e00 100644 --- a/src/diffusers/models/vae.py +++ b/src/diffusers/models/vae.py @@ -534,7 +534,7 @@ class VQModel(ModelMixin, ConfigMixin): quant = self.post_quant_conv(quant) dec = self.decoder(quant) return dec - + def forward(self, x): h = self.encode(x) dec = self.decode(h)