Upload liquid_diffusion/__init__.py

liquid_diffusion/__init__.py

@@ -0,0 +1,20 @@
+# LiquidDiffusion: Novel Attention-Free Image Generation with Liquid Neural Networks
+#
+# Based on:
+#   CfC Networks [Hasani et al., Nature MI 2022]
+#   LiquidTAD parallel relaxation [arxiv 2604.18274]
+#   USM architecture [arxiv 2504.13499]
+#   Rectified Flow [Liu et al., ICLR 2023]
+
+from liquid_diffusion.model import (
+    LiquidDiffusionUNet,
+    liquid_diffusion_tiny,
+    liquid_diffusion_small,
+    liquid_diffusion_base,
+    liquid_diffusion_large,
+)
+from liquid_diffusion.trainer import (
+    RectifiedFlowTrainer,
+    ImageDataset,
+    get_cosine_schedule_with_warmup,
+)