awjuliani / pytorch-diffusion Goto Github PK

Hi,

Thanks for the code. I faced a problem when I tried to generate the gif.

Cannot handle this data type: (1, 1, 1), |u1

How could I solve this problem? Thanks!

First of all - Thanks for the great implementation, which is very readable and simple to use and reach quick results with.

I noticed you're using the self attention in a different manner than the one written by Ho et al - they used the self attention in the 16x16 resolution layers in the network (both for 32x32 and 256x256 inputs) while you used it in the 8x8 resolution layers in the U-Net. Did you just happen to use it there / was there some guiding logic behind it?