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)