File size: 738 Bytes
34bede7 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | """
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 |