1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00
This commit is contained in:
patil-suraj
2022-06-29 12:34:28 +02:00
parent bae04ea9d8
commit 976173a4bf

View File

@@ -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)