chuanyangjin / fast-DiT

Fast Diffusion Models with Transformers
Other
735 stars 94 forks source link

how to condition on an image? #8

Open amirshamaei opened 8 months ago

amirshamaei commented 8 months ago

I was wondering if it is possible to condition the input on an image? since the scale size is (Bdim) so how we can scale the input (BLdim) using an image. Should I patch the image to size of (B, pp , dim)?