add Distribute to fluxion layers's __init__.py

This commit is contained in:
Cédric Deltheil 2023-09-01 10:36:23 +02:00 committed by Cédric Deltheil
parent 9cf622a6e2
commit 92cdf19eae

View file

@ -24,6 +24,7 @@ from refiners.fluxion.layers.chain import (
UseContext, UseContext,
SetContext, SetContext,
Parallel, Parallel,
Distribute,
Passthrough, Passthrough,
Breakpoint, Breakpoint,
Concatenate, Concatenate,
@ -71,6 +72,7 @@ __all__ = [
"UseContext", "UseContext",
"SetContext", "SetContext",
"Parallel", "Parallel",
"Distribute",
"Passthrough", "Passthrough",
"Breakpoint", "Breakpoint",
"Concatenate", "Concatenate",