liquid_diffusion/__init__.py

# 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,
)