export the magnitude preserving temporal layers

lucidrains · lucidrains · commit 25b40cdb445d · 2024-02-14T09:41:26.000-08:00
diff --git a/lumiere_pytorch/__init__.py b/lumiere_pytorch/__init__.py
@@ -9,5 +9,9 @@
 from lumiere_pytorch.lumiere import Lumiere
 
 from lumiere_pytorch.mp_lumiere import (
-    MPLumiere
+    MPLumiere,
+    MPConvolutionInflationBlock,
+    MPAttentionInflationBlock,
+    MPTemporalDownsample,
+    MPTemporalUpsample,
 )
diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 setup(
   name = 'lumiere-pytorch',
   packages = find_packages(exclude=[]),
-  version = '0.0.18',
+  version = '0.0.19',
   license='MIT',
   description = 'Lumiere',
   author = 'Phil Wang',
