Skip to content

GitLab

Commit 976173a4 authored by patil-suraj's avatar patil-suraj
Browse files

style

parent bae04ea9
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/diffusers/models/vae.py
View file @ 976173a4
...@@ -534,7 +534,7 @@ class VQModel(ModelMixin, ConfigMixin): ...@@ -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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment