Mirrors
/
diffusers
mirror of https://github.com/ShivamShrirao/diffusers.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

style

This commit is contained in:
patil-suraj 2022-06-29 12:34:28 +02:00
parent bae04ea9d8
commit 976173a4bf
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/diffusers/models/vae.py
View File

@ -534,7 +534,7 @@ class VQModel(ModelMixin, ConfigMixin):
quant = self.post_quant_conv(quant) quant = self.post_quant_conv(quant)
dec = self.decoder(quant) dec = self.decoder(quant)
return dec return dec
def forward(self, x): def forward(self, x):
h = self.encode(x) h = self.encode(x)
dec = self.decode(h) dec = self.decode(h)