style
This commit is contained in:
parent
bae04ea9d8
commit
976173a4bf
|
@ -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)
|
||||
|
|
Loading…
Reference in New Issue