mirror of
https://github.com/huggingface/diffusers.git
synced 2026-01-29 07:22:12 +03:00
update
This commit is contained in:
@@ -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
|
||||
|
||||
@@ -21,9 +21,9 @@ import torch
|
||||
import torch.nn.functional as F
|
||||
from torch import nn
|
||||
|
||||
from ....configuration_utils import ConfigMixin, register_to_config
|
||||
from ....models import ModelMixin
|
||||
from ....utils import BaseOutput
|
||||
from ...configuration_utils import ConfigMixin, register_to_config
|
||||
from ...models import ModelMixin
|
||||
from .camera import create_pan_cameras
|
||||
|
||||
|
||||
|
||||
@@ -22,6 +22,7 @@ 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
|
||||
@@ -39,7 +40,6 @@ from ....utils import (
|
||||
unscale_lora_layers,
|
||||
)
|
||||
from ....utils.torch_utils import randn_tensor
|
||||
from ...configuration_utils import FrozenDict
|
||||
from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin
|
||||
from ..stable_diffusion import StableDiffusionPipelineOutput
|
||||
from ..stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
||||
|
||||
@@ -7,13 +7,13 @@ 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 ...configuration_utils import FrozenDict
|
||||
from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin
|
||||
from . import StableDiffusionSafePipelineOutput
|
||||
from .safety_checker import SafeStableDiffusionSafetyChecker
|
||||
|
||||
@@ -15,8 +15,8 @@
|
||||
import torch
|
||||
from torch import nn
|
||||
|
||||
from ...configuration_utils import ConfigMixin, register_to_config
|
||||
from ...models import ModelMixin
|
||||
from ....configuration_utils import ConfigMixin, register_to_config
|
||||
from ....models import ModelMixin
|
||||
|
||||
|
||||
class UnCLIPTextProjModel(ModelMixin, ConfigMixin):
|
||||
|
||||
@@ -6,8 +6,8 @@ from torch import nn
|
||||
from transformers import GPT2Config, GPT2LMHeadModel
|
||||
from transformers.modeling_utils import ModuleUtilsMixin
|
||||
|
||||
from ...configuration_utils import ConfigMixin, register_to_config
|
||||
from ...models import ModelMixin
|
||||
from ....configuration_utils import ConfigMixin, register_to_config
|
||||
from ....models import ModelMixin
|
||||
|
||||
|
||||
# Modified from ClipCaptionModel in https://github.com/thu-ml/unidiffuser/blob/main/libs/caption_decoder.py
|
||||
|
||||
@@ -4,14 +4,14 @@ from typing import Optional, Union
|
||||
import torch
|
||||
from torch import nn
|
||||
|
||||
from ....configuration_utils import ConfigMixin, register_to_config
|
||||
from ....models import ModelMixin
|
||||
from ....models.attention import FeedForward
|
||||
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
|
||||
from ...configuration_utils import ConfigMixin, register_to_config
|
||||
from ...models import ModelMixin
|
||||
from ...models.attention import FeedForward
|
||||
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
|
||||
|
||||
|
||||
logger = logging.get_logger(__name__) # pylint: disable=invalid-name
|
||||
|
||||
Reference in New Issue
Block a user