Skip to content

GitLab

Unverified Commit 6be43bd8 authored by townwish4git's avatar townwish4git Committed by GitHub
Browse files

Fix AsymmetricAutoencoderKL forward (#8378)

parent dc89434b
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/diffusers/models/autoencoders/autoencoder_asym_kl.py
View file @ 6be43bd8
...@@ -176,7 +176,7 @@ class AsymmetricAutoencoderKL(ModelMixin, ConfigMixin): ...@@ -176,7 +176,7 @@ class AsymmetricAutoencoderKL(ModelMixin, ConfigMixin):
z = posterior.sample(generator=generator) z = posterior.sample(generator=generator)
else: else:
z = posterior.mode() z = posterior.mode()
dec = self.decode(z, sample, mask).sample dec = self.decode(z, generator, sample, mask).sample
if not return_dict: if not return_dict:
return (dec,) return (dec,)
......
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