Training on unconditional image generation creates colorized images

[esizikova-fda]

Describe the bug

Hi, I'm trying to follow the tutorial from unconditional image generation on my own dataset, and I'm getting weirdly colored images. I originally thought it was due to RGB/BGR channel order, but I've switched it around and got the same result. Do you have any suggestions of how to fix it?

Reproduction

NA

Logs

System Info

NA

Who can help?

No response