Unverified Commit 3552279a authored by Dhruv Nair's avatar Dhruv Nair Committed by GitHub
Browse files

[Modular] Add experimental feature warning for Modular Diffusers (#12127)

update
parent f8ba5cd7
...@@ -7,9 +7,15 @@ from ..utils import ( ...@@ -7,9 +7,15 @@ from ..utils import (
get_objects_from_module, get_objects_from_module,
is_torch_available, is_torch_available,
is_transformers_available, is_transformers_available,
logging,
) )
logger = logging.get_logger(__name__)
logger.warning(
"Modular Diffusers is currently an experimental feature under active development. The API is subject to breaking changes in future releases."
)
# These modules contain pipelines from multiple libraries/frameworks # These modules contain pipelines from multiple libraries/frameworks
_dummy_objects = {} _dummy_objects = {}
_import_structure = {} _import_structure = {}
...@@ -61,17 +67,8 @@ if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT: ...@@ -61,17 +67,8 @@ if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
PipelineState, PipelineState,
SequentialPipelineBlocks, SequentialPipelineBlocks,
) )
from .modular_pipeline_utils import ( from .modular_pipeline_utils import ComponentSpec, ConfigSpec, InputParam, InsertableDict, OutputParam
ComponentSpec, from .stable_diffusion_xl import StableDiffusionXLAutoBlocks, StableDiffusionXLModularPipeline
ConfigSpec,
InputParam,
InsertableDict,
OutputParam,
)
from .stable_diffusion_xl import (
StableDiffusionXLAutoBlocks,
StableDiffusionXLModularPipeline,
)
from .wan import WanAutoBlocks, WanModularPipeline from .wan import WanAutoBlocks, WanModularPipeline
else: else:
import sys import sys
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment