search

facebookresearch / DiT

Official PyTorch Implementation of "Scalable Diffusion Models with Transformers"
Other
6.37k stars 570 forks source link

Some generated pictures looks like wrong, have u ever meet this ? #20

Closed heepengpeng closed 1 year ago

heepengpeng commented 1 year ago

sampleLcBDmJrC

wpeebles commented 1 year ago

Hi @heepengpeng. I haven't seen samples like this before :) is this sampling from the pre-trained DiT models? would you mind sharing a code snippet/ more details?

heepengpeng commented 1 year ago

Hi @heepengpeng. I haven't seen samples like this before :) is this sampling from the pre-trained DiT models? would you mind sharing a code snippet/ more details?

https://github.com/heepengpeng/minddiffusion/blob/main/vision/DiT/mainV2.py code is here. I use the the pre-trained DiT models. I rewrite the code by another AI framework MindSpore . https://github.com/heepengpeng/minddiffusion/blob/main/vision/DiT/mainV2.py

wpeebles commented 1 year ago

So it's a bit hard to tell if there's an issue from one sample, but if there is it's likely with the mindspore codebase. If you can reproduce the issue with the PyTorch codebase, let me know and I'll try to take a look.