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:41:45 +05:30
parent b797186e84
commit 19afac19fa
7 changed files with 16 additions and 16 deletions

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

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

View File

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

View File

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

View File

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

View File

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

View File

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