| """ | |
| LiquidFlow: A Lightweight Image Generator combining | |
| Liquid Neural Networks (CfC) + Mamba-2 SSD + Physics-Informed Regularization | |
| Based on: | |
| - CfC: "Closed-form Continuous-time Neural Networks" (Hasani et al., Nature MI 2022) | |
| - Mamba-2: "Transformers are SSMs" (Dao & Gu, 2024) | |
| - PINN Diffusion: "Physics-Informed Diffusion Models" (Bastek & Sun, ICLR 2025) | |
| - DiMSUM: "Diffusion Mamba" (NeurIPS 2024) | |
| """ | |
| from .cfc_cell import CfCCell, CfCBlock | |
| from .mamba2_ssd import Mamba2SSD, Mamba2Block | |
| from .liquid_flow_block import LiquidFlowBlock, LiquidFlowStage | |
| from .generator import LiquidFlowGenerator | |
| from .vae_wrapper import TAESDWrapper, SDVAEWrapper | |
| from .physics_loss import PhysicsRegularizer | |
| from .trainer import LiquidFlowTrainer |