From 6be43bd855772f5ffb064bc7c6049f578b3856f8 Mon Sep 17 00:00:00 2001 From: townwish4git <143256262+townwish4git@users.noreply.github.com> Date: Tue, 4 Jun 2024 11:25:11 +0800 Subject: [PATCH] Fix AsymmetricAutoencoderKL forward (#8378) --- src/diffusers/models/autoencoders/autoencoder_asym_kl.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/models/autoencoders/autoencoder_asym_kl.py b/src/diffusers/models/autoencoders/autoencoder_asym_kl.py index 85b4195077..3f4d46557b 100644 --- a/src/diffusers/models/autoencoders/autoencoder_asym_kl.py +++ b/src/diffusers/models/autoencoders/autoencoder_asym_kl.py @@ -176,7 +176,7 @@ class AsymmetricAutoencoderKL(ModelMixin, ConfigMixin): z = posterior.sample(generator=generator) else: z = posterior.mode() - dec = self.decode(z, sample, mask).sample + dec = self.decode(z, generator, sample, mask).sample if not return_dict: return (dec,)