A DiT block with parallel linear layers as described in https://arxiv.org/abs/2302.05442 and adapted modulation interface.

Checkpoint resume or save hook. Args: interval (int): Save interval, by epoch. save_best (bool): Save the best checkpoint by a me

reuse the ldm code

support to load the components dynamicly. create and load model when run this model at the first time.

A DiT block with parallel linear layers as described in https://arxiv.org/abs/2302.05442 and adapted modulation interface.

A DiT block with parallel linear layers as described in https://arxiv.org/abs/2302.05442 and adapted modulation interface.

Transformer backbone Diffusion model with RoPE.

A DiT block with parallel linear layers as described in https://arxiv.org/abs/2302.05442 and adapted modulation interface.

Uses the OpenCLIP transformer encoder for text