mirror of
https://github.com/huggingface/diffusers.git
synced 2026-01-29 07:22:12 +03:00
update
This commit is contained in:
@@ -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,
|
||||
|
||||
@@ -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
|
||||
|
||||
@@ -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
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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():
|
||||
|
||||
@@ -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():
|
||||
|
||||
@@ -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():
|
||||
|
||||
@@ -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():
|
||||
|
||||
@@ -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
|
||||
|
||||
@@ -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
|
||||
|
||||
@@ -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
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
@@ -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
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
|
||||
@@ -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():
|
||||
|
||||
@@ -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
|
||||
|
||||
@@ -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
|
||||
|
||||
@@ -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
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
|
||||
@@ -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):
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
|
||||
@@ -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
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user