Upload liquid_flow/__init__.py

liquid_flow/__init__.py

@@ -1,5 +1,5 @@
 """
-LiquidFlow: A Lightweight Image Generator combining
+LiquidFlow: Lightweight Image Generator combining
 Liquid Neural Networks (CfC) + Mamba-2 SSD + Physics-Informed Regularization
 
 Based on:
@@ -11,8 +11,7 @@ Based on:
 
 from .cfc_cell import CfCCell, CfCBlock
 from .mamba2_ssd import Mamba2SSD, Mamba2Block
-from .liquid_flow_block import LiquidFlowBlock, LiquidFlowStage
-from .generator import LiquidFlowGenerator
+from .liquid_flow_block import LiquidMambaBlock, LiquidFlowStage, LiquidFlowBackbone
+from .generator import LiquidFlowGenerator, create_liquidflow
 from .vae_wrapper import TAESDWrapper, SDVAEWrapper
-from .physics_loss import PhysicsRegularizer
-from .trainer import LiquidFlowTrainer
+from .physics_loss import PhysicsRegularizer, DDIMEstimator