Black image: The result of ddim_sample is black, while p_sample_loop can generate fine images.

[GuHuangAI]

p_sample_loop:

ddim_sample:

What happened?

[GuHuangAI]

I find the balck image is not caused by ddim_sample but by ema model, why does the ema model generate wrong image? I set the ema_decay to 0.995.

[BEbillionaireUSD]

I find the balck image is not caused by ddim_sample but by ema model, why does the ema model generate wrong image? I set the ema_decay to 0.995.

How did you save it? Did you close ema?

[BEbillionaireUSD]

I find the balck image is not caused by ddim_sample but by ema model, why does the ema model generate wrong image? I set the ema_decay to 0.995.

How did you save it? Did you close ema? I have the same problem. Could you please share your solution? Thanks:)

[GuHuangAI]

@cherylLbt Yeah, i closed ema. You can replace self.ema.ema_model.sample by self.model.module.sample in denoising_diffusion.py.

[BEbillionaireUSD]

@cherylLbt Yeah, i closed ema. You can replace self.ema.ema_model.sample by self.model.module.sample in denoising_diffusion.py.

Many thanks for your reply!!🙏