Upload liquid_flow/__init__.py
Browse files- liquid_flow/__init__.py +18 -0
liquid_flow/__init__.py
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
"""
|
| 2 |
+
LiquidFlow: A Lightweight Image Generator combining
|
| 3 |
+
Liquid Neural Networks (CfC) + Mamba-2 SSD + Physics-Informed Regularization
|
| 4 |
+
|
| 5 |
+
Based on:
|
| 6 |
+
- CfC: "Closed-form Continuous-time Neural Networks" (Hasani et al., Nature MI 2022)
|
| 7 |
+
- Mamba-2: "Transformers are SSMs" (Dao & Gu, 2024)
|
| 8 |
+
- PINN Diffusion: "Physics-Informed Diffusion Models" (Bastek & Sun, ICLR 2025)
|
| 9 |
+
- DiMSUM: "Diffusion Mamba" (NeurIPS 2024)
|
| 10 |
+
"""
|
| 11 |
+
|
| 12 |
+
from .cfc_cell import CfCCell, CfCBlock
|
| 13 |
+
from .mamba2_ssd import Mamba2SSD, Mamba2Block
|
| 14 |
+
from .liquid_flow_block import LiquidFlowBlock, LiquidFlowStage
|
| 15 |
+
from .generator import LiquidFlowGenerator
|
| 16 |
+
from .vae_wrapper import TAESDWrapper, SDVAEWrapper
|
| 17 |
+
from .physics_loss import PhysicsRegularizer
|
| 18 |
+
from .trainer import LiquidFlowTrainer
|