code for DiT block with In-Context Conditioning

facebookresearch / DiT

Official PyTorch Implementation of "Scalable Diffusion Models with Transformers"

Other

6.17k stars 544 forks source link

code for DiT block with In-Context Conditioning #39

Open moonbaby9 opened 1 year ago

moonbaby9 commented 1 year ago

Thanks for your wonderful work! I learned that there were three designs for DiT block in your paper . However, this public code just contains code for adaLN Zero block design. I would appreciate it if you could make the code for the other two structures(In-Context conditioning and cross-attention block) public.

mayank64ce commented 1 year ago

I am looking for the same.

manhuii commented 1 month ago

Same here.