| # LiquidDiffusion: Novel Attention-Free Image Generation with Liquid Neural Networks | |
| # | |
| # Based on: | |
| # CfC Networks [Hasani et al., Nature MI 2022] | |
| # LiquidTAD parallel relaxation [arxiv 2604.18274] | |
| # USM architecture [arxiv 2504.13499] | |
| # Rectified Flow [Liu et al., ICLR 2023] | |
| from liquid_diffusion.model import ( | |
| LiquidDiffusionUNet, | |
| liquid_diffusion_tiny, | |
| liquid_diffusion_small, | |
| liquid_diffusion_base, | |
| liquid_diffusion_large, | |
| ) | |
| from liquid_diffusion.trainer import ( | |
| RectifiedFlowTrainer, | |
| ImageDataset, | |
| get_cosine_schedule_with_warmup, | |
| ) | |