Training on unconditional image generation creates colorized images

### 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

```shell

```

### System Info

NA

### Who can help?

_No response_