add Distribute to fluxion layers's __init__.py

finegrain-ai · Sep 1, 2023 · 92cdf19 · 92cdf19
1 parent 9cf622a
commit 92cdf19
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/refiners/fluxion/layers/__init__.py b/src/refiners/fluxion/layers/__init__.py
@@ -24,6 +24,7 @@
     UseContext,
     SetContext,
     Parallel,
+    Distribute,
     Passthrough,
     Breakpoint,
     Concatenate,
@@ -71,6 +72,7 @@
     "UseContext",
     "SetContext",
     "Parallel",
+    "Distribute",
     "Passthrough",
     "Breakpoint",
     "Concatenate",