1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00
This commit is contained in:
DN6
2025-04-17 00:39:19 +05:30
parent c68db8014f
commit b797186e84
46 changed files with 79 additions and 73 deletions

View File

@@ -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,

View File

@@ -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

View File

@@ -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

View File

@@ -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

View File

@@ -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,

View File

@@ -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():

View File

@@ -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():

View File

@@ -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():

View File

@@ -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():

View File

@@ -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

View File

@@ -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

View File

@@ -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

View File

@@ -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

View File

@@ -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

View File

@@ -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

View File

@@ -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

View File

@@ -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

View File

@@ -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

View File

@@ -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

View File

@@ -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

View File

@@ -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

View File

@@ -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

View File

@@ -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

View File

@@ -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

View File

@@ -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

View File

@@ -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

View File

@@ -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

View File

@@ -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

View File

@@ -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

View File

@@ -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

View File

@@ -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

View File

@@ -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

View File

@@ -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

View File

@@ -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

View File

@@ -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

View File

@@ -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

View File

@@ -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():

View File

@@ -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

View File

@@ -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

View File

@@ -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

View File

@@ -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

View File

@@ -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):

View File

@@ -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

View File

@@ -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

View File

@@ -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

View File

@@ -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