Skip to content

Commit 3552279

Browse files
authored
[Modular] Add experimental feature warning for Modular Diffusers (#12127)
update
1 parent f8ba5cd commit 3552279

File tree

1 file changed

+8
-11
lines changed

1 file changed

+8
-11
lines changed

src/diffusers/modular_pipelines/__init__.py

Lines changed: 8 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -7,9 +7,15 @@
77
get_objects_from_module,
88
is_torch_available,
99
is_transformers_available,
10+
logging,
1011
)
1112

1213

14+
logger = logging.get_logger(__name__)
15+
logger.warning(
16+
"Modular Diffusers is currently an experimental feature under active development. The API is subject to breaking changes in future releases."
17+
)
18+
1319
# These modules contain pipelines from multiple libraries/frameworks
1420
_dummy_objects = {}
1521
_import_structure = {}
@@ -61,17 +67,8 @@
6167
PipelineState,
6268
SequentialPipelineBlocks,
6369
)
64-
from .modular_pipeline_utils import (
65-
ComponentSpec,
66-
ConfigSpec,
67-
InputParam,
68-
InsertableDict,
69-
OutputParam,
70-
)
71-
from .stable_diffusion_xl import (
72-
StableDiffusionXLAutoBlocks,
73-
StableDiffusionXLModularPipeline,
74-
)
70+
from .modular_pipeline_utils import ComponentSpec, ConfigSpec, InputParam, InsertableDict, OutputParam
71+
from .stable_diffusion_xl import StableDiffusionXLAutoBlocks, StableDiffusionXLModularPipeline
7572
from .wan import WanAutoBlocks, WanModularPipeline
7673
else:
7774
import sys

0 commit comments

Comments
 (0)