Fix AsymmetricAutoencoderKL forward (huggingface#8378)

sirpaddy · Jun 4, 2024 · 6be43bd · 6be43bd
1 parent dc89434
commit 6be43bd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/diffusers/models/autoencoders/autoencoder_asym_kl.py b/src/diffusers/models/autoencoders/autoencoder_asym_kl.py
@@ -176,7 +176,7 @@ def forward(
             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,)