krystv commited on
Commit
34bede7
·
verified ·
1 Parent(s): 3330459

Upload liquid_flow/__init__.py

Browse files
Files changed (1) hide show
  1. 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