diff --git a/src/diffusers/guiders/adaptive_projected_guidance.py b/src/diffusers/guiders/adaptive_projected_guidance.py index 83e93c15ff..ef2f3f2c84 100644 --- a/src/diffusers/guiders/adaptive_projected_guidance.py +++ b/src/diffusers/guiders/adaptive_projected_guidance.py @@ -20,7 +20,7 @@ import torch from .guider_utils import BaseGuidance, rescale_noise_cfg if TYPE_CHECKING: - from ..pipelines.modular_pipeline import BlockState + from ..modular_pipelines.modular_pipeline import BlockState class AdaptiveProjectedGuidance(BaseGuidance): diff --git a/src/diffusers/guiders/auto_guidance.py b/src/diffusers/guiders/auto_guidance.py index 8bb6083781..791cc582ad 100644 --- a/src/diffusers/guiders/auto_guidance.py +++ b/src/diffusers/guiders/auto_guidance.py @@ -22,7 +22,7 @@ from ..hooks.layer_skip import _apply_layer_skip_hook from .guider_utils import BaseGuidance, rescale_noise_cfg if TYPE_CHECKING: - from ..pipelines.modular_pipeline import BlockState + from ..modular_pipelines.modular_pipeline import BlockState class AutoGuidance(BaseGuidance): diff --git a/src/diffusers/guiders/classifier_free_guidance.py b/src/diffusers/guiders/classifier_free_guidance.py index 429392e3f9..a459e51cd0 100644 --- a/src/diffusers/guiders/classifier_free_guidance.py +++ b/src/diffusers/guiders/classifier_free_guidance.py @@ -20,7 +20,7 @@ import torch from .guider_utils import BaseGuidance, rescale_noise_cfg if TYPE_CHECKING: - from ..pipelines.modular_pipeline import BlockState + from ..modular_pipelines.modular_pipeline import BlockState class ClassifierFreeGuidance(BaseGuidance): diff --git a/src/diffusers/guiders/classifier_free_zero_star_guidance.py b/src/diffusers/guiders/classifier_free_zero_star_guidance.py index 220a95e54a..a722f26050 100644 --- a/src/diffusers/guiders/classifier_free_zero_star_guidance.py +++ b/src/diffusers/guiders/classifier_free_zero_star_guidance.py @@ -20,7 +20,7 @@ import torch from .guider_utils import BaseGuidance, rescale_noise_cfg if TYPE_CHECKING: - from ..pipelines.modular_pipeline import BlockState + from ..modular_pipelines.modular_pipeline import BlockState class ClassifierFreeZeroStarGuidance(BaseGuidance): diff --git a/src/diffusers/guiders/guider_utils.py b/src/diffusers/guiders/guider_utils.py index df544c955f..e8e873f5c8 100644 --- a/src/diffusers/guiders/guider_utils.py +++ b/src/diffusers/guiders/guider_utils.py @@ -20,7 +20,7 @@ from ..utils import get_logger if TYPE_CHECKING: - from ..pipelines.modular_pipeline import BlockState + from ..modular_pipelines.modular_pipeline import BlockState logger = get_logger(__name__) # pylint: disable=invalid-name @@ -171,7 +171,7 @@ class BaseGuidance: Returns: `BlockState`: The prepared batch of data. """ - from ..pipelines.modular_pipeline import BlockState + from ..modular_pipelines.modular_pipeline import BlockState if input_fields is None: raise ValueError("Input fields cannot be None. Please pass `input_fields` to `prepare_inputs` or call `set_input_fields` before preparing inputs.") diff --git a/src/diffusers/guiders/skip_layer_guidance.py b/src/diffusers/guiders/skip_layer_guidance.py index 56dae19036..7c19f6391f 100644 --- a/src/diffusers/guiders/skip_layer_guidance.py +++ b/src/diffusers/guiders/skip_layer_guidance.py @@ -22,7 +22,7 @@ from ..hooks.layer_skip import _apply_layer_skip_hook from .guider_utils import BaseGuidance, rescale_noise_cfg if TYPE_CHECKING: - from ..pipelines.modular_pipeline import BlockState + from ..modular_pipelines.modular_pipeline import BlockState class SkipLayerGuidance(BaseGuidance): diff --git a/src/diffusers/guiders/smoothed_energy_guidance.py b/src/diffusers/guiders/smoothed_energy_guidance.py index c215cb0afd..3986da913f 100644 --- a/src/diffusers/guiders/smoothed_energy_guidance.py +++ b/src/diffusers/guiders/smoothed_energy_guidance.py @@ -22,7 +22,7 @@ from ..hooks.smoothed_energy_guidance_utils import SmoothedEnergyGuidanceConfig, from .guider_utils import BaseGuidance, rescale_noise_cfg if TYPE_CHECKING: - from ..pipelines.modular_pipeline import BlockState + from ..modular_pipelines.modular_pipeline import BlockState class SmoothedEnergyGuidance(BaseGuidance): diff --git a/src/diffusers/guiders/tangential_classifier_free_guidance.py b/src/diffusers/guiders/tangential_classifier_free_guidance.py index 9fa8f94541..017693fd9f 100644 --- a/src/diffusers/guiders/tangential_classifier_free_guidance.py +++ b/src/diffusers/guiders/tangential_classifier_free_guidance.py @@ -20,7 +20,7 @@ import torch from .guider_utils import BaseGuidance, rescale_noise_cfg if TYPE_CHECKING: - from ..pipelines.modular_pipeline import BlockState + from ..modular_pipelines.modular_pipeline import BlockState class TangentialClassifierFreeGuidance(BaseGuidance):