From b797186e8487432028b2cc874eafcbdc33c10b3f Mon Sep 17 00:00:00 2001 From: DN6 Date: Thu, 17 Apr 2025 00:39:19 +0530 Subject: [PATCH] update --- .../pipelines/deprecated/__init__.py | 22 ++++++++++--------- .../alt_diffusion/pipeline_alt_diffusion.py | 2 +- .../pipeline_alt_diffusion_img2img.py | 2 +- .../pipeline_audio_diffusion.py | 2 +- .../deprecated/audioldm/pipeline_audioldm.py | 6 ++--- .../pipeline_dance_diffusion.py | 2 +- .../i2vgen_xl/pipeline_i2vgen_xl.py | 2 +- .../deprecated/latte/pipeline_latte.py | 2 +- .../deprecated/musicldm/pipeline_musicldm.py | 2 +- .../paint_by_example/image_encoder.py | 2 +- .../pipeline_paint_by_example.py | 2 +- .../pipelines/deprecated/pia/pipeline_pia.py | 3 ++- .../pipeline_semantic_stable_diffusion.py | 4 ++-- .../deprecated/shap_e/pipeline_shap_e.py | 2 +- .../shap_e/pipeline_shap_e_img2img.py | 2 +- .../pipelines/deprecated/shap_e/renderer.py | 2 +- .../pipeline_spectrogram_diffusion.py | 2 +- ...line_stable_diffusion_attend_and_excite.py | 2 +- .../pipeline_stable_diffusion_diffedit.py | 4 ++-- .../pipeline_stable_diffusion_gligen.py | 2 +- ...line_stable_diffusion_gligen_text_image.py | 2 +- .../pipeline_stable_diffusion_k_diffusion.py | 2 +- ...ipeline_stable_diffusion_xl_k_diffusion.py | 2 +- .../pipeline_stable_diffusion_ldm3d.py | 9 ++++++-- .../pipeline_stable_diffusion_panorama.py | 2 +- .../pipeline_stable_diffusion_safe.py | 4 ++-- .../pipeline_stable_diffusion_sag.py | 2 +- .../pipeline_cycle_diffusion.py | 2 +- ...ne_onnx_stable_diffusion_inpaint_legacy.py | 2 +- ...ipeline_stable_diffusion_inpaint_legacy.py | 2 +- ...pipeline_stable_diffusion_model_editing.py | 2 +- .../pipeline_stable_diffusion_paradigms.py | 2 +- .../pipeline_stable_diffusion_pix2pix_zero.py | 2 +- .../pipeline_text_to_video_synth.py | 2 +- .../pipeline_text_to_video_synth_img2img.py | 2 +- .../pipeline_text_to_video_zero.py | 2 +- .../pipeline_text_to_video_zero_sdxl.py | 2 +- .../deprecated/unidiffuser/modeling_uvit.py | 2 +- .../unidiffuser/pipeline_unidiffuser.py | 10 ++++----- .../pipeline_versatile_diffusion.py | 2 +- .../wuerstchen/modeling_paella_vq_model.py | 8 +++---- .../wuerstchen/modeling_wuerstchen_common.py | 2 +- .../modeling_wuerstchen_diffnext.py | 4 ++-- .../wuerstchen/modeling_wuerstchen_prior.py | 8 +++---- .../pipeline_wuerstchen_combined.py | 2 +- .../wuerstchen/pipeline_wuerstchen_prior.py | 2 +- 46 files changed, 79 insertions(+), 73 deletions(-) diff --git a/src/diffusers/pipelines/deprecated/__init__.py b/src/diffusers/pipelines/deprecated/__init__.py index 6464752f5b..84cbd55dc1 100644 --- a/src/diffusers/pipelines/deprecated/__init__.py +++ b/src/diffusers/pipelines/deprecated/__init__.py @@ -1,8 +1,8 @@ -from typing import TYPE_CHECKING import functools import inspect -import warnings import sys +import warnings +from typing import TYPE_CHECKING from ...utils import ( DIFFUSERS_SLOW_IMPORT, @@ -13,24 +13,26 @@ from ...utils import ( is_note_seq_available, is_torch_available, is_transformers_available, + logging, ) +logger = logging.get_logger(__name__) + + # Custom Lazy Module for deprecated pipelines that shows a warning class _DeprecatedLazyModule(_LazyModule): """ - Module class that surfaces all objects but only performs associated imports when the objects are requested, - and shows deprecation warnings when any of its attributes are accessed. + Module class that surfaces all objects but only performs associated imports when the objects are requested, and + shows deprecation warnings when any of its attributes are accessed. """ def __getattr__(self, name): # Regular attribute access - first check if it's supposed to be loaded if name in self._modules or name in self._class_to_module: # Only warn for actual pipeline components, not utility functions - warnings.warn( - f"{name} is deprecated and will no longer be maintained.", - FutureWarning, - stacklevel=2, + logger.warning( + f"{name} is deprecated and will no longer be maintained or receive future updates.", ) # Use the standard lazy module behavior to load the attribute @@ -184,7 +186,7 @@ if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT: from .paint_by_example import PaintByExamplePipeline from .pia import PIAPipeline from .semantic_stable_diffusion import SemanticStableDiffusionPipeline - from .shap_e import ShapEPipeline, ShapEImg2ImgPipeline + from .shap_e import ShapEImg2ImgPipeline, ShapEPipeline from .stable_diffusion_attend_and_excite import StableDiffusionAttendAndExcitePipeline from .stable_diffusion_diffedit import StableDiffusionDiffEditPipeline from .stable_diffusion_gligen import StableDiffusionGLIGENPipeline, StableDiffusionGLIGENTextImagePipeline @@ -208,7 +210,7 @@ if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT: TextToVideoZeroPipeline, TextToVideoZeroSDXLPipeline, ) - from .unclip import UnCLIPPipeline, UnCLIPImageVariationPipeline + from .unclip import UnCLIPImageVariationPipeline, UnCLIPPipeline from .unidiffuser import UniDiffuserPipeline from .versatile_diffusion import ( VersatileDiffusionDualGuidedPipeline, diff --git a/src/diffusers/pipelines/deprecated/alt_diffusion/pipeline_alt_diffusion.py b/src/diffusers/pipelines/deprecated/alt_diffusion/pipeline_alt_diffusion.py index 6fc9e1ec39..4dce54808d 100644 --- a/src/diffusers/pipelines/deprecated/alt_diffusion/pipeline_alt_diffusion.py +++ b/src/diffusers/pipelines/deprecated/alt_diffusion/pipeline_alt_diffusion.py @@ -39,7 +39,7 @@ from ....utils import ( unscale_lora_layers, ) from ....utils.torch_utils import randn_tensor -from ..pipeline_utils import DiffusionPipeline, StableDiffusionMixin +from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin from ..stable_diffusion.safety_checker import StableDiffusionSafetyChecker from .modeling_roberta_series import RobertaSeriesModelWithTransformation from .pipeline_output import AltDiffusionPipelineOutput diff --git a/src/diffusers/pipelines/deprecated/alt_diffusion/pipeline_alt_diffusion_img2img.py b/src/diffusers/pipelines/deprecated/alt_diffusion/pipeline_alt_diffusion_img2img.py index 563c81d1ce..7538ab0098 100644 --- a/src/diffusers/pipelines/deprecated/alt_diffusion/pipeline_alt_diffusion_img2img.py +++ b/src/diffusers/pipelines/deprecated/alt_diffusion/pipeline_alt_diffusion_img2img.py @@ -42,7 +42,7 @@ from ....utils import ( unscale_lora_layers, ) from ....utils.torch_utils import randn_tensor -from ..pipeline_utils import DiffusionPipeline, StableDiffusionMixin +from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin from ..stable_diffusion.safety_checker import StableDiffusionSafetyChecker from .modeling_roberta_series import RobertaSeriesModelWithTransformation from .pipeline_output import AltDiffusionPipelineOutput diff --git a/src/diffusers/pipelines/deprecated/audio_diffusion/pipeline_audio_diffusion.py b/src/diffusers/pipelines/deprecated/audio_diffusion/pipeline_audio_diffusion.py index 663e474344..47044e050a 100644 --- a/src/diffusers/pipelines/deprecated/audio_diffusion/pipeline_audio_diffusion.py +++ b/src/diffusers/pipelines/deprecated/audio_diffusion/pipeline_audio_diffusion.py @@ -23,7 +23,7 @@ from PIL import Image from ....models import AutoencoderKL, UNet2DConditionModel from ....schedulers import DDIMScheduler, DDPMScheduler from ....utils.torch_utils import randn_tensor -from ..pipeline_utils import AudioPipelineOutput, BaseOutput, DiffusionPipeline, ImagePipelineOutput +from ...pipeline_utils import AudioPipelineOutput, BaseOutput, DiffusionPipeline, ImagePipelineOutput from .mel import Mel diff --git a/src/diffusers/pipelines/deprecated/audioldm/pipeline_audioldm.py b/src/diffusers/pipelines/deprecated/audioldm/pipeline_audioldm.py index 307325e7e7..279b992569 100644 --- a/src/diffusers/pipelines/deprecated/audioldm/pipeline_audioldm.py +++ b/src/diffusers/pipelines/deprecated/audioldm/pipeline_audioldm.py @@ -24,7 +24,7 @@ from ....models import AutoencoderKL, UNet2DConditionModel from ....schedulers import KarrasDiffusionSchedulers from ....utils import is_torch_xla_available, logging, replace_example_docstring from ....utils.torch_utils import randn_tensor -from ..pipeline_utils import AudioPipelineOutput, DiffusionPipeline, StableDiffusionMixin +from ...pipeline_utils import AudioPipelineOutput, DiffusionPipeline, StableDiffusionMixin if is_torch_xla_available(): @@ -96,9 +96,7 @@ class AudioLDMPipeline(DiffusionPipeline, StableDiffusionMixin): logger.warning(f"{self.__class__.__name__} is deprecated and will no longer be actively maintained") - logger.warning( - f"{self.__class__.__name__} is deprecated and will no longer be actively maintained" - ) + logger.warning(f"{self.__class__.__name__} is deprecated and will no longer be actively maintained") self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/deprecated/dance_diffusion/pipeline_dance_diffusion.py b/src/diffusers/pipelines/deprecated/dance_diffusion/pipeline_dance_diffusion.py index 0075eb8b31..aab0337ed5 100644 --- a/src/diffusers/pipelines/deprecated/dance_diffusion/pipeline_dance_diffusion.py +++ b/src/diffusers/pipelines/deprecated/dance_diffusion/pipeline_dance_diffusion.py @@ -21,7 +21,7 @@ from ....models import UNet1DModel from ....schedulers import SchedulerMixin from ....utils import is_torch_xla_available, logging from ....utils.torch_utils import randn_tensor -from ..pipeline_utils import AudioPipelineOutput, DiffusionPipeline +from ...pipeline_utils import AudioPipelineOutput, DiffusionPipeline if is_torch_xla_available(): diff --git a/src/diffusers/pipelines/deprecated/i2vgen_xl/pipeline_i2vgen_xl.py b/src/diffusers/pipelines/deprecated/i2vgen_xl/pipeline_i2vgen_xl.py index e34a4ea26f..df8e6b195d 100644 --- a/src/diffusers/pipelines/deprecated/i2vgen_xl/pipeline_i2vgen_xl.py +++ b/src/diffusers/pipelines/deprecated/i2vgen_xl/pipeline_i2vgen_xl.py @@ -33,7 +33,7 @@ from ....utils import ( ) from ....utils.torch_utils import randn_tensor from ....video_processor import VideoProcessor -from ..pipeline_utils import DiffusionPipeline, StableDiffusionMixin +from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin if is_torch_xla_available(): diff --git a/src/diffusers/pipelines/deprecated/latte/pipeline_latte.py b/src/diffusers/pipelines/deprecated/latte/pipeline_latte.py index dfc1022a0b..0117f0e7b6 100644 --- a/src/diffusers/pipelines/deprecated/latte/pipeline_latte.py +++ b/src/diffusers/pipelines/deprecated/latte/pipeline_latte.py @@ -25,7 +25,6 @@ from transformers import T5EncoderModel, T5Tokenizer from ....callbacks import MultiPipelineCallbacks, PipelineCallback from ....models import AutoencoderKL, LatteTransformer3DModel -from ...pipeline_utils import DiffusionPipeline from ....schedulers import KarrasDiffusionSchedulers from ....utils import ( BACKENDS_MAPPING, @@ -39,6 +38,7 @@ from ....utils import ( ) from ....utils.torch_utils import is_compiled_module, randn_tensor from ....video_processor import VideoProcessor +from ...pipeline_utils import DiffusionPipeline if is_torch_xla_available(): diff --git a/src/diffusers/pipelines/deprecated/musicldm/pipeline_musicldm.py b/src/diffusers/pipelines/deprecated/musicldm/pipeline_musicldm.py index b1753756ff..e2d2dc4c33 100644 --- a/src/diffusers/pipelines/deprecated/musicldm/pipeline_musicldm.py +++ b/src/diffusers/pipelines/deprecated/musicldm/pipeline_musicldm.py @@ -36,7 +36,7 @@ from ....utils import ( replace_example_docstring, ) from ....utils.torch_utils import randn_tensor -from ..pipeline_utils import AudioPipelineOutput, DiffusionPipeline, StableDiffusionMixin +from ...pipeline_utils import AudioPipelineOutput, DiffusionPipeline, StableDiffusionMixin if is_librosa_available(): diff --git a/src/diffusers/pipelines/deprecated/paint_by_example/image_encoder.py b/src/diffusers/pipelines/deprecated/paint_by_example/image_encoder.py index 76482de4a2..30f0ea666e 100644 --- a/src/diffusers/pipelines/deprecated/paint_by_example/image_encoder.py +++ b/src/diffusers/pipelines/deprecated/paint_by_example/image_encoder.py @@ -15,8 +15,8 @@ import torch from torch import nn from transformers import CLIPPreTrainedModel, CLIPVisionModel -from ...models.attention import BasicTransformerBlock from ....utils import logging +from ...models.attention import BasicTransformerBlock logger = logging.get_logger(__name__) # pylint: disable=invalid-name diff --git a/src/diffusers/pipelines/deprecated/paint_by_example/pipeline_paint_by_example.py b/src/diffusers/pipelines/deprecated/paint_by_example/pipeline_paint_by_example.py index 8454948a04..223925f1ca 100644 --- a/src/diffusers/pipelines/deprecated/paint_by_example/pipeline_paint_by_example.py +++ b/src/diffusers/pipelines/deprecated/paint_by_example/pipeline_paint_by_example.py @@ -25,7 +25,7 @@ from ....models import AutoencoderKL, UNet2DConditionModel from ....schedulers import DDIMScheduler, LMSDiscreteScheduler, PNDMScheduler from ....utils import deprecate, is_torch_xla_available, logging from ....utils.torch_utils import randn_tensor -from ..pipeline_utils import DiffusionPipeline, StableDiffusionMixin +from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin from ..stable_diffusion import StableDiffusionPipelineOutput from ..stable_diffusion.safety_checker import StableDiffusionSafetyChecker from .image_encoder import PaintByExampleImageEncoder diff --git a/src/diffusers/pipelines/deprecated/pia/pipeline_pia.py b/src/diffusers/pipelines/deprecated/pia/pipeline_pia.py index 566c794b78..6401829a8b 100644 --- a/src/diffusers/pipelines/deprecated/pia/pipeline_pia.py +++ b/src/diffusers/pipelines/deprecated/pia/pipeline_pia.py @@ -50,8 +50,9 @@ from ....utils import ( ) from ....utils.torch_utils import randn_tensor from ....video_processor import VideoProcessor +from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin from ..free_init_utils import FreeInitMixin -from ..pipeline_utils import DiffusionPipeline, StableDiffusionMixin + if is_torch_xla_available(): import torch_xla.core.xla_model as xm diff --git a/src/diffusers/pipelines/deprecated/semantic_stable_diffusion/pipeline_semantic_stable_diffusion.py b/src/diffusers/pipelines/deprecated/semantic_stable_diffusion/pipeline_semantic_stable_diffusion.py index 53f5c73014..559639cdff 100644 --- a/src/diffusers/pipelines/deprecated/semantic_stable_diffusion/pipeline_semantic_stable_diffusion.py +++ b/src/diffusers/pipelines/deprecated/semantic_stable_diffusion/pipeline_semantic_stable_diffusion.py @@ -7,11 +7,11 @@ from transformers import CLIPImageProcessor, CLIPTextModel, CLIPTokenizer from ....image_processor import VaeImageProcessor from ....models import AutoencoderKL, UNet2DConditionModel -from ...pipelines.stable_diffusion.safety_checker import StableDiffusionSafetyChecker from ....schedulers import KarrasDiffusionSchedulers from ....utils import deprecate, is_torch_xla_available, logging from ....utils.torch_utils import randn_tensor -from ..pipeline_utils import DiffusionPipeline, StableDiffusionMixin +from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin +from ...stable_diffusion.safety_checker import StableDiffusionSafetyChecker from .pipeline_output import SemanticStableDiffusionPipelineOutput diff --git a/src/diffusers/pipelines/deprecated/shap_e/pipeline_shap_e.py b/src/diffusers/pipelines/deprecated/shap_e/pipeline_shap_e.py index 6819107453..1018cefa81 100644 --- a/src/diffusers/pipelines/deprecated/shap_e/pipeline_shap_e.py +++ b/src/diffusers/pipelines/deprecated/shap_e/pipeline_shap_e.py @@ -30,7 +30,7 @@ from ....utils import ( replace_example_docstring, ) from ....utils.torch_utils import randn_tensor -from ..pipeline_utils import DiffusionPipeline +from ...pipeline_utils import DiffusionPipeline from .renderer import ShapERenderer diff --git a/src/diffusers/pipelines/deprecated/shap_e/pipeline_shap_e_img2img.py b/src/diffusers/pipelines/deprecated/shap_e/pipeline_shap_e_img2img.py index 620bde9f01..3b3e1e8467 100644 --- a/src/diffusers/pipelines/deprecated/shap_e/pipeline_shap_e_img2img.py +++ b/src/diffusers/pipelines/deprecated/shap_e/pipeline_shap_e_img2img.py @@ -29,7 +29,7 @@ from ....utils import ( replace_example_docstring, ) from ....utils.torch_utils import randn_tensor -from ..pipeline_utils import DiffusionPipeline +from ...pipeline_utils import DiffusionPipeline from .renderer import ShapERenderer diff --git a/src/diffusers/pipelines/deprecated/shap_e/renderer.py b/src/diffusers/pipelines/deprecated/shap_e/renderer.py index 2510fe870a..52b7534405 100644 --- a/src/diffusers/pipelines/deprecated/shap_e/renderer.py +++ b/src/diffusers/pipelines/deprecated/shap_e/renderer.py @@ -21,9 +21,9 @@ import torch import torch.nn.functional as F from torch import nn +from ....utils import BaseOutput from ...configuration_utils import ConfigMixin, register_to_config from ...models import ModelMixin -from ....utils import BaseOutput from .camera import create_pan_cameras diff --git a/src/diffusers/pipelines/deprecated/spectrogram_diffusion/pipeline_spectrogram_diffusion.py b/src/diffusers/pipelines/deprecated/spectrogram_diffusion/pipeline_spectrogram_diffusion.py index 242add16f3..b8ac8e1416 100644 --- a/src/diffusers/pipelines/deprecated/spectrogram_diffusion/pipeline_spectrogram_diffusion.py +++ b/src/diffusers/pipelines/deprecated/spectrogram_diffusion/pipeline_spectrogram_diffusion.py @@ -28,7 +28,7 @@ from ....utils.torch_utils import randn_tensor if is_onnx_available(): from ...onnx_utils import OnnxRuntimeModel -from ..pipeline_utils import AudioPipelineOutput, DiffusionPipeline +from ...pipeline_utils import AudioPipelineOutput, DiffusionPipeline from .continuous_encoder import SpectrogramContEncoder from .notes_encoder import SpectrogramNotesEncoder diff --git a/src/diffusers/pipelines/deprecated/stable_diffusion_attend_and_excite/pipeline_stable_diffusion_attend_and_excite.py b/src/diffusers/pipelines/deprecated/stable_diffusion_attend_and_excite/pipeline_stable_diffusion_attend_and_excite.py index eb09b478b3..535bece706 100644 --- a/src/diffusers/pipelines/deprecated/stable_diffusion_attend_and_excite/pipeline_stable_diffusion_attend_and_excite.py +++ b/src/diffusers/pipelines/deprecated/stable_diffusion_attend_and_excite/pipeline_stable_diffusion_attend_and_excite.py @@ -37,7 +37,7 @@ from ....utils import ( unscale_lora_layers, ) from ....utils.torch_utils import randn_tensor -from ..pipeline_utils import DiffusionPipeline, StableDiffusionMixin +from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin from ..stable_diffusion import StableDiffusionPipelineOutput from ..stable_diffusion.safety_checker import StableDiffusionSafetyChecker diff --git a/src/diffusers/pipelines/deprecated/stable_diffusion_diffedit/pipeline_stable_diffusion_diffedit.py b/src/diffusers/pipelines/deprecated/stable_diffusion_diffedit/pipeline_stable_diffusion_diffedit.py index 9206a392ab..272a7fc7b7 100644 --- a/src/diffusers/pipelines/deprecated/stable_diffusion_diffedit/pipeline_stable_diffusion_diffedit.py +++ b/src/diffusers/pipelines/deprecated/stable_diffusion_diffedit/pipeline_stable_diffusion_diffedit.py @@ -22,7 +22,6 @@ import torch from packaging import version from transformers import CLIPImageProcessor, CLIPTextModel, CLIPTokenizer -from ...configuration_utils import FrozenDict from ....image_processor import VaeImageProcessor from ....loaders import StableDiffusionLoraLoaderMixin, TextualInversionLoaderMixin from ....models import AutoencoderKL, UNet2DConditionModel @@ -40,7 +39,8 @@ from ....utils import ( unscale_lora_layers, ) from ....utils.torch_utils import randn_tensor -from ..pipeline_utils import DiffusionPipeline, StableDiffusionMixin +from ...configuration_utils import FrozenDict +from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin from ..stable_diffusion import StableDiffusionPipelineOutput from ..stable_diffusion.safety_checker import StableDiffusionSafetyChecker diff --git a/src/diffusers/pipelines/deprecated/stable_diffusion_gligen/pipeline_stable_diffusion_gligen.py b/src/diffusers/pipelines/deprecated/stable_diffusion_gligen/pipeline_stable_diffusion_gligen.py index a7d64a579c..f3068de32b 100644 --- a/src/diffusers/pipelines/deprecated/stable_diffusion_gligen/pipeline_stable_diffusion_gligen.py +++ b/src/diffusers/pipelines/deprecated/stable_diffusion_gligen/pipeline_stable_diffusion_gligen.py @@ -36,7 +36,7 @@ from ....utils import ( unscale_lora_layers, ) from ....utils.torch_utils import randn_tensor -from ..pipeline_utils import DiffusionPipeline, StableDiffusionMixin +from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin from ..stable_diffusion import StableDiffusionPipelineOutput from ..stable_diffusion.safety_checker import StableDiffusionSafetyChecker diff --git a/src/diffusers/pipelines/deprecated/stable_diffusion_gligen/pipeline_stable_diffusion_gligen_text_image.py b/src/diffusers/pipelines/deprecated/stable_diffusion_gligen/pipeline_stable_diffusion_gligen_text_image.py index 5423d6bca9..0e6e8d999d 100644 --- a/src/diffusers/pipelines/deprecated/stable_diffusion_gligen/pipeline_stable_diffusion_gligen_text_image.py +++ b/src/diffusers/pipelines/deprecated/stable_diffusion_gligen/pipeline_stable_diffusion_gligen_text_image.py @@ -41,7 +41,7 @@ from ....utils import ( unscale_lora_layers, ) from ....utils.torch_utils import randn_tensor -from ..pipeline_utils import DiffusionPipeline, StableDiffusionMixin +from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin from ..stable_diffusion import StableDiffusionPipelineOutput from ..stable_diffusion.clip_image_project_model import CLIPImageProjection from ..stable_diffusion.safety_checker import StableDiffusionSafetyChecker diff --git a/src/diffusers/pipelines/deprecated/stable_diffusion_k_diffusion/pipeline_stable_diffusion_k_diffusion.py b/src/diffusers/pipelines/deprecated/stable_diffusion_k_diffusion/pipeline_stable_diffusion_k_diffusion.py index fd9052317d..79c6aaf7dd 100755 --- a/src/diffusers/pipelines/deprecated/stable_diffusion_k_diffusion/pipeline_stable_diffusion_k_diffusion.py +++ b/src/diffusers/pipelines/deprecated/stable_diffusion_k_diffusion/pipeline_stable_diffusion_k_diffusion.py @@ -42,7 +42,7 @@ from ....utils import ( unscale_lora_layers, ) from ....utils.torch_utils import randn_tensor -from ..pipeline_utils import DiffusionPipeline, StableDiffusionMixin +from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin from ..stable_diffusion import StableDiffusionPipelineOutput, StableDiffusionSafetyChecker diff --git a/src/diffusers/pipelines/deprecated/stable_diffusion_k_diffusion/pipeline_stable_diffusion_xl_k_diffusion.py b/src/diffusers/pipelines/deprecated/stable_diffusion_k_diffusion/pipeline_stable_diffusion_xl_k_diffusion.py index 776ed5bddb..68cbe29d68 100644 --- a/src/diffusers/pipelines/deprecated/stable_diffusion_k_diffusion/pipeline_stable_diffusion_xl_k_diffusion.py +++ b/src/diffusers/pipelines/deprecated/stable_diffusion_k_diffusion/pipeline_stable_diffusion_xl_k_diffusion.py @@ -48,7 +48,7 @@ from ....utils import ( unscale_lora_layers, ) from ....utils.torch_utils import randn_tensor -from ..pipeline_utils import DiffusionPipeline, StableDiffusionMixin +from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin from ..stable_diffusion_xl.pipeline_output import StableDiffusionXLPipelineOutput diff --git a/src/diffusers/pipelines/deprecated/stable_diffusion_ldm3d/pipeline_stable_diffusion_ldm3d.py b/src/diffusers/pipelines/deprecated/stable_diffusion_ldm3d/pipeline_stable_diffusion_ldm3d.py index 578523309a..f2a920c0f0 100644 --- a/src/diffusers/pipelines/deprecated/stable_diffusion_ldm3d/pipeline_stable_diffusion_ldm3d.py +++ b/src/diffusers/pipelines/deprecated/stable_diffusion_ldm3d/pipeline_stable_diffusion_ldm3d.py @@ -22,7 +22,12 @@ import torch from transformers import CLIPImageProcessor, CLIPTextModel, CLIPTokenizer, CLIPVisionModelWithProjection from ....image_processor import PipelineImageInput, VaeImageProcessorLDM3D -from ....loaders import FromSingleFileMixin, IPAdapterMixin, StableDiffusionLoraLoaderMixin, TextualInversionLoaderMixin +from ....loaders import ( + FromSingleFileMixin, + IPAdapterMixin, + StableDiffusionLoraLoaderMixin, + TextualInversionLoaderMixin, +) from ....models import AutoencoderKL, ImageProjection, UNet2DConditionModel from ....models.lora import adjust_lora_scale_text_encoder from ....schedulers import KarrasDiffusionSchedulers @@ -37,7 +42,7 @@ from ....utils import ( unscale_lora_layers, ) from ....utils.torch_utils import randn_tensor -from ..pipeline_utils import DiffusionPipeline, StableDiffusionMixin +from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin from ..stable_diffusion.safety_checker import StableDiffusionSafetyChecker diff --git a/src/diffusers/pipelines/deprecated/stable_diffusion_panorama/pipeline_stable_diffusion_panorama.py b/src/diffusers/pipelines/deprecated/stable_diffusion_panorama/pipeline_stable_diffusion_panorama.py index e09433f308..42bbadd56f 100644 --- a/src/diffusers/pipelines/deprecated/stable_diffusion_panorama/pipeline_stable_diffusion_panorama.py +++ b/src/diffusers/pipelines/deprecated/stable_diffusion_panorama/pipeline_stable_diffusion_panorama.py @@ -33,7 +33,7 @@ from ....utils import ( unscale_lora_layers, ) from ....utils.torch_utils import randn_tensor -from ..pipeline_utils import DiffusionPipeline, StableDiffusionMixin +from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin from ..stable_diffusion import StableDiffusionPipelineOutput from ..stable_diffusion.safety_checker import StableDiffusionSafetyChecker diff --git a/src/diffusers/pipelines/deprecated/stable_diffusion_safe/pipeline_stable_diffusion_safe.py b/src/diffusers/pipelines/deprecated/stable_diffusion_safe/pipeline_stable_diffusion_safe.py index 837670b88c..34e32b6b31 100644 --- a/src/diffusers/pipelines/deprecated/stable_diffusion_safe/pipeline_stable_diffusion_safe.py +++ b/src/diffusers/pipelines/deprecated/stable_diffusion_safe/pipeline_stable_diffusion_safe.py @@ -7,14 +7,14 @@ import torch from packaging import version from transformers import CLIPImageProcessor, CLIPTextModel, CLIPTokenizer, CLIPVisionModelWithProjection -from ...configuration_utils import FrozenDict from ....image_processor import PipelineImageInput from ....loaders import IPAdapterMixin from ....models import AutoencoderKL, ImageProjection, UNet2DConditionModel from ....schedulers import KarrasDiffusionSchedulers from ....utils import deprecate, is_torch_xla_available, logging from ....utils.torch_utils import randn_tensor -from ..pipeline_utils import DiffusionPipeline, StableDiffusionMixin +from ...configuration_utils import FrozenDict +from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin from . import StableDiffusionSafePipelineOutput from .safety_checker import SafeStableDiffusionSafetyChecker diff --git a/src/diffusers/pipelines/deprecated/stable_diffusion_sag/pipeline_stable_diffusion_sag.py b/src/diffusers/pipelines/deprecated/stable_diffusion_sag/pipeline_stable_diffusion_sag.py index 07986bc709..26a13e3b62 100644 --- a/src/diffusers/pipelines/deprecated/stable_diffusion_sag/pipeline_stable_diffusion_sag.py +++ b/src/diffusers/pipelines/deprecated/stable_diffusion_sag/pipeline_stable_diffusion_sag.py @@ -34,7 +34,7 @@ from ....utils import ( unscale_lora_layers, ) from ....utils.torch_utils import randn_tensor -from ..pipeline_utils import DiffusionPipeline, StableDiffusionMixin +from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin from ..stable_diffusion import StableDiffusionPipelineOutput from ..stable_diffusion.safety_checker import StableDiffusionSafetyChecker diff --git a/src/diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_cycle_diffusion.py b/src/diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_cycle_diffusion.py index 9193932841..4bf5e6f156 100644 --- a/src/diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_cycle_diffusion.py +++ b/src/diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_cycle_diffusion.py @@ -29,7 +29,7 @@ from ....models.lora import adjust_lora_scale_text_encoder from ....schedulers import DDIMScheduler from ....utils import PIL_INTERPOLATION, USE_PEFT_BACKEND, deprecate, logging, scale_lora_layers, unscale_lora_layers from ....utils.torch_utils import randn_tensor -from ..pipeline_utils import DiffusionPipeline +from ...pipeline_utils import DiffusionPipeline from ..stable_diffusion.pipeline_output import StableDiffusionPipelineOutput from ..stable_diffusion.safety_checker import StableDiffusionSafetyChecker diff --git a/src/diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_onnx_stable_diffusion_inpaint_legacy.py b/src/diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_onnx_stable_diffusion_inpaint_legacy.py index f19f914425..6296ea935c 100644 --- a/src/diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_onnx_stable_diffusion_inpaint_legacy.py +++ b/src/diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_onnx_stable_diffusion_inpaint_legacy.py @@ -10,7 +10,7 @@ from ....configuration_utils import FrozenDict from ....schedulers import DDIMScheduler, LMSDiscreteScheduler, PNDMScheduler from ....utils import deprecate, logging from ...onnx_utils import ORT_TO_NP_TYPE, OnnxRuntimeModel -from ..pipeline_utils import DiffusionPipeline +from ...pipeline_utils import DiffusionPipeline from ..stable_diffusion.pipeline_output import StableDiffusionPipelineOutput diff --git a/src/diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_inpaint_legacy.py b/src/diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_inpaint_legacy.py index 3d087cf20f..12a5c84034 100644 --- a/src/diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_inpaint_legacy.py +++ b/src/diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_inpaint_legacy.py @@ -29,7 +29,7 @@ from ....models.lora import adjust_lora_scale_text_encoder from ....schedulers import KarrasDiffusionSchedulers from ....utils import PIL_INTERPOLATION, USE_PEFT_BACKEND, deprecate, logging, scale_lora_layers, unscale_lora_layers from ....utils.torch_utils import randn_tensor -from ..pipeline_utils import DiffusionPipeline +from ...pipeline_utils import DiffusionPipeline from ..stable_diffusion import StableDiffusionPipelineOutput from ..stable_diffusion.safety_checker import StableDiffusionSafetyChecker diff --git a/src/diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_model_editing.py b/src/diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_model_editing.py index 125a0c6338..9e66be6a4a 100644 --- a/src/diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_model_editing.py +++ b/src/diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_model_editing.py @@ -26,7 +26,7 @@ from ....schedulers import PNDMScheduler from ....schedulers.scheduling_utils import SchedulerMixin from ....utils import USE_PEFT_BACKEND, deprecate, logging, scale_lora_layers, unscale_lora_layers from ....utils.torch_utils import randn_tensor -from ..pipeline_utils import DiffusionPipeline, StableDiffusionMixin +from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin from ..stable_diffusion.pipeline_output import StableDiffusionPipelineOutput from ..stable_diffusion.safety_checker import StableDiffusionSafetyChecker diff --git a/src/diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_paradigms.py b/src/diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_paradigms.py index 98d78d2909..b69a07becc 100644 --- a/src/diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_paradigms.py +++ b/src/diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_paradigms.py @@ -32,7 +32,7 @@ from ....utils import ( unscale_lora_layers, ) from ....utils.torch_utils import randn_tensor -from ..pipeline_utils import DiffusionPipeline, StableDiffusionMixin +from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin from ..stable_diffusion.pipeline_output import StableDiffusionPipelineOutput from ..stable_diffusion.safety_checker import StableDiffusionSafetyChecker diff --git a/src/diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_pix2pix_zero.py b/src/diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_pix2pix_zero.py index b635211e92..072a097fed 100644 --- a/src/diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_pix2pix_zero.py +++ b/src/diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_pix2pix_zero.py @@ -46,7 +46,7 @@ from ....utils import ( unscale_lora_layers, ) from ....utils.torch_utils import randn_tensor -from ..pipeline_utils import DiffusionPipeline, StableDiffusionMixin +from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin from ..stable_diffusion.pipeline_output import StableDiffusionPipelineOutput from ..stable_diffusion.safety_checker import StableDiffusionSafetyChecker diff --git a/src/diffusers/pipelines/deprecated/text_to_video_synthesis/pipeline_text_to_video_synth.py b/src/diffusers/pipelines/deprecated/text_to_video_synthesis/pipeline_text_to_video_synth.py index 379c1ca8c6..da7f9d3e6b 100644 --- a/src/diffusers/pipelines/deprecated/text_to_video_synthesis/pipeline_text_to_video_synth.py +++ b/src/diffusers/pipelines/deprecated/text_to_video_synthesis/pipeline_text_to_video_synth.py @@ -33,7 +33,7 @@ from ....utils import ( ) from ....utils.torch_utils import randn_tensor from ....video_processor import VideoProcessor -from ..pipeline_utils import DiffusionPipeline, StableDiffusionMixin +from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin from . import TextToVideoSDPipelineOutput diff --git a/src/diffusers/pipelines/deprecated/text_to_video_synthesis/pipeline_text_to_video_synth_img2img.py b/src/diffusers/pipelines/deprecated/text_to_video_synthesis/pipeline_text_to_video_synth_img2img.py index 63ba18e476..de17cdd847 100644 --- a/src/diffusers/pipelines/deprecated/text_to_video_synthesis/pipeline_text_to_video_synth_img2img.py +++ b/src/diffusers/pipelines/deprecated/text_to_video_synthesis/pipeline_text_to_video_synth_img2img.py @@ -34,7 +34,7 @@ from ....utils import ( ) from ....utils.torch_utils import randn_tensor from ....video_processor import VideoProcessor -from ..pipeline_utils import DiffusionPipeline, StableDiffusionMixin +from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin from . import TextToVideoSDPipelineOutput diff --git a/src/diffusers/pipelines/deprecated/text_to_video_synthesis/pipeline_text_to_video_zero.py b/src/diffusers/pipelines/deprecated/text_to_video_synthesis/pipeline_text_to_video_zero.py index 6a67f4c480..098d520986 100644 --- a/src/diffusers/pipelines/deprecated/text_to_video_synthesis/pipeline_text_to_video_zero.py +++ b/src/diffusers/pipelines/deprecated/text_to_video_synthesis/pipeline_text_to_video_zero.py @@ -24,7 +24,7 @@ from ....utils import ( unscale_lora_layers, ) from ....utils.torch_utils import randn_tensor -from ..pipeline_utils import DiffusionPipeline, StableDiffusionMixin +from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin from ..stable_diffusion import StableDiffusionSafetyChecker diff --git a/src/diffusers/pipelines/deprecated/text_to_video_synthesis/pipeline_text_to_video_zero_sdxl.py b/src/diffusers/pipelines/deprecated/text_to_video_synthesis/pipeline_text_to_video_zero_sdxl.py index 5581f80d35..e9b1aae997 100644 --- a/src/diffusers/pipelines/deprecated/text_to_video_synthesis/pipeline_text_to_video_zero_sdxl.py +++ b/src/diffusers/pipelines/deprecated/text_to_video_synthesis/pipeline_text_to_video_zero_sdxl.py @@ -35,7 +35,7 @@ from ....utils import ( unscale_lora_layers, ) from ....utils.torch_utils import randn_tensor -from ..pipeline_utils import DiffusionPipeline, StableDiffusionMixin +from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin if is_invisible_watermark_available(): diff --git a/src/diffusers/pipelines/deprecated/unidiffuser/modeling_uvit.py b/src/diffusers/pipelines/deprecated/unidiffuser/modeling_uvit.py index 0b3175e2f2..75ad205fdd 100644 --- a/src/diffusers/pipelines/deprecated/unidiffuser/modeling_uvit.py +++ b/src/diffusers/pipelines/deprecated/unidiffuser/modeling_uvit.py @@ -4,6 +4,7 @@ from typing import Optional, Union import torch from torch import nn +from ....utils import logging from ...configuration_utils import ConfigMixin, register_to_config from ...models import ModelMixin from ...models.attention import FeedForward @@ -11,7 +12,6 @@ from ...models.attention_processor import Attention from ...models.embeddings import TimestepEmbedding, Timesteps, get_2d_sincos_pos_embed from ...models.modeling_outputs import Transformer2DModelOutput from ...models.normalization import AdaLayerNorm -from ....utils import logging logger = logging.get_logger(__name__) # pylint: disable=invalid-name diff --git a/src/diffusers/pipelines/deprecated/unidiffuser/pipeline_unidiffuser.py b/src/diffusers/pipelines/deprecated/unidiffuser/pipeline_unidiffuser.py index 40d30e9442..d03baf9587 100644 --- a/src/diffusers/pipelines/deprecated/unidiffuser/pipeline_unidiffuser.py +++ b/src/diffusers/pipelines/deprecated/unidiffuser/pipeline_unidiffuser.py @@ -28,7 +28,7 @@ from ....utils import ( ) from ....utils.outputs import BaseOutput from ....utils.torch_utils import randn_tensor -from ..pipeline_utils import DiffusionPipeline +from ...pipeline_utils import DiffusionPipeline from .modeling_text_decoder import UniDiffuserTextDecoder from .modeling_uvit import UniDiffuserModel @@ -225,7 +225,7 @@ class UniDiffuserPipeline(DiffusionPipeline): return mode - # Copied from ..pipeline_utils.StableDiffusionMixin.enable_vae_slicing + # Copied from ...pipeline_utils.StableDiffusionMixin.enable_vae_slicing def enable_vae_slicing(self): r""" Enable sliced VAE decoding. When this option is enabled, the VAE will split the input tensor in slices to @@ -233,7 +233,7 @@ class UniDiffuserPipeline(DiffusionPipeline): """ self.vae.enable_slicing() - # Copied from ..pipeline_utils.StableDiffusionMixin.disable_vae_slicing + # Copied from ...pipeline_utils.StableDiffusionMixin.disable_vae_slicing def disable_vae_slicing(self): r""" Disable sliced VAE decoding. If `enable_vae_slicing` was previously enabled, this method will go back to @@ -241,7 +241,7 @@ class UniDiffuserPipeline(DiffusionPipeline): """ self.vae.disable_slicing() - # Copied from ..pipeline_utils.StableDiffusionMixin.enable_vae_tiling + # Copied from ...pipeline_utils.StableDiffusionMixin.enable_vae_tiling def enable_vae_tiling(self): r""" Enable tiled VAE decoding. When this option is enabled, the VAE will split the input tensor into tiles to @@ -250,7 +250,7 @@ class UniDiffuserPipeline(DiffusionPipeline): """ self.vae.enable_tiling() - # Copied from ..pipeline_utils.StableDiffusionMixin.disable_vae_tiling + # Copied from ...pipeline_utils.StableDiffusionMixin.disable_vae_tiling def disable_vae_tiling(self): r""" Disable tiled VAE decoding. If `enable_vae_tiling` was previously enabled, this method will go back to diff --git a/src/diffusers/pipelines/deprecated/versatile_diffusion/pipeline_versatile_diffusion.py b/src/diffusers/pipelines/deprecated/versatile_diffusion/pipeline_versatile_diffusion.py index 126dddd253..4fb437958a 100644 --- a/src/diffusers/pipelines/deprecated/versatile_diffusion/pipeline_versatile_diffusion.py +++ b/src/diffusers/pipelines/deprecated/versatile_diffusion/pipeline_versatile_diffusion.py @@ -8,7 +8,7 @@ from transformers import CLIPImageProcessor, CLIPTextModel, CLIPTokenizer, CLIPV from ....models import AutoencoderKL, UNet2DConditionModel from ....schedulers import KarrasDiffusionSchedulers from ....utils import logging -from ..pipeline_utils import DiffusionPipeline +from ...pipeline_utils import DiffusionPipeline from .pipeline_versatile_diffusion_dual_guided import VersatileDiffusionDualGuidedPipeline from .pipeline_versatile_diffusion_image_variation import VersatileDiffusionImageVariationPipeline from .pipeline_versatile_diffusion_text_to_image import VersatileDiffusionTextToImagePipeline diff --git a/src/diffusers/pipelines/deprecated/wuerstchen/modeling_paella_vq_model.py b/src/diffusers/pipelines/deprecated/wuerstchen/modeling_paella_vq_model.py index ab8d259dc1..bf5cc35101 100644 --- a/src/diffusers/pipelines/deprecated/wuerstchen/modeling_paella_vq_model.py +++ b/src/diffusers/pipelines/deprecated/wuerstchen/modeling_paella_vq_model.py @@ -18,10 +18,10 @@ from typing import Union import torch import torch.nn as nn -from ...configuration_utils import ConfigMixin, register_to_config -from ...models.autoencoders.vae import DecoderOutput, VectorQuantizer -from ...models.modeling_utils import ModelMixin -from ...models.vq_model import VQEncoderOutput +from ....configuration_utils import ConfigMixin, register_to_config +from ....models.autoencoders.vae import DecoderOutput, VectorQuantizer +from ....models.modeling_utils import ModelMixin +from ....models.vq_model import VQEncoderOutput from ....utils.accelerate_utils import apply_forward_hook diff --git a/src/diffusers/pipelines/deprecated/wuerstchen/modeling_wuerstchen_common.py b/src/diffusers/pipelines/deprecated/wuerstchen/modeling_wuerstchen_common.py index 73e71b3076..7645a5579c 100644 --- a/src/diffusers/pipelines/deprecated/wuerstchen/modeling_wuerstchen_common.py +++ b/src/diffusers/pipelines/deprecated/wuerstchen/modeling_wuerstchen_common.py @@ -1,7 +1,7 @@ import torch import torch.nn as nn -from ...models.attention_processor import Attention +from ....models.attention_processor import Attention class WuerstchenLayerNorm(nn.LayerNorm): diff --git a/src/diffusers/pipelines/deprecated/wuerstchen/modeling_wuerstchen_diffnext.py b/src/diffusers/pipelines/deprecated/wuerstchen/modeling_wuerstchen_diffnext.py index 6c06cc0e73..7157287f00 100644 --- a/src/diffusers/pipelines/deprecated/wuerstchen/modeling_wuerstchen_diffnext.py +++ b/src/diffusers/pipelines/deprecated/wuerstchen/modeling_wuerstchen_diffnext.py @@ -19,8 +19,8 @@ import numpy as np import torch import torch.nn as nn -from ...configuration_utils import ConfigMixin, register_to_config -from ...models.modeling_utils import ModelMixin +from ....configuration_utils import ConfigMixin, register_to_config +from ....models.modeling_utils import ModelMixin from .modeling_wuerstchen_common import AttnBlock, GlobalResponseNorm, TimestepBlock, WuerstchenLayerNorm diff --git a/src/diffusers/pipelines/deprecated/wuerstchen/modeling_wuerstchen_prior.py b/src/diffusers/pipelines/deprecated/wuerstchen/modeling_wuerstchen_prior.py index 9863c506d7..ef34795d3d 100644 --- a/src/diffusers/pipelines/deprecated/wuerstchen/modeling_wuerstchen_prior.py +++ b/src/diffusers/pipelines/deprecated/wuerstchen/modeling_wuerstchen_prior.py @@ -19,16 +19,16 @@ from typing import Dict, Union import torch import torch.nn as nn -from ...configuration_utils import ConfigMixin, register_to_config -from ...loaders import PeftAdapterMixin, UNet2DConditionLoadersMixin -from ...models.attention_processor import ( +from ....configuration_utils import ConfigMixin, register_to_config +from ....loaders import PeftAdapterMixin, UNet2DConditionLoadersMixin +from ....models.attention_processor import ( ADDED_KV_ATTENTION_PROCESSORS, CROSS_ATTENTION_PROCESSORS, AttentionProcessor, AttnAddedKVProcessor, AttnProcessor, ) -from ...models.modeling_utils import ModelMixin +from ....models.modeling_utils import ModelMixin from .modeling_wuerstchen_common import AttnBlock, ResBlock, TimestepBlock, WuerstchenLayerNorm diff --git a/src/diffusers/pipelines/deprecated/wuerstchen/pipeline_wuerstchen_combined.py b/src/diffusers/pipelines/deprecated/wuerstchen/pipeline_wuerstchen_combined.py index 4d1bc7abf5..2cf57b2a10 100644 --- a/src/diffusers/pipelines/deprecated/wuerstchen/pipeline_wuerstchen_combined.py +++ b/src/diffusers/pipelines/deprecated/wuerstchen/pipeline_wuerstchen_combined.py @@ -18,7 +18,7 @@ from transformers import CLIPTextModel, CLIPTokenizer from ....schedulers import DDPMWuerstchenScheduler from ....utils import deprecate, replace_example_docstring -from ..pipeline_utils import DiffusionPipeline +from ...pipeline_utils import DiffusionPipeline from .modeling_paella_vq_model import PaellaVQModel from .modeling_wuerstchen_diffnext import WuerstchenDiffNeXt from .modeling_wuerstchen_prior import WuerstchenPrior diff --git a/src/diffusers/pipelines/deprecated/wuerstchen/pipeline_wuerstchen_prior.py b/src/diffusers/pipelines/deprecated/wuerstchen/pipeline_wuerstchen_prior.py index 7cef8a3be0..3c086c1878 100644 --- a/src/diffusers/pipelines/deprecated/wuerstchen/pipeline_wuerstchen_prior.py +++ b/src/diffusers/pipelines/deprecated/wuerstchen/pipeline_wuerstchen_prior.py @@ -24,7 +24,7 @@ from ....loaders import StableDiffusionLoraLoaderMixin from ....schedulers import DDPMWuerstchenScheduler from ....utils import BaseOutput, deprecate, is_torch_xla_available, logging, replace_example_docstring from ....utils.torch_utils import randn_tensor -from ..pipeline_utils import DiffusionPipeline +from ...pipeline_utils import DiffusionPipeline from .modeling_wuerstchen_prior import WuerstchenPrior