Update pipeline.py
Browse files- pipeline.py +13 -13
pipeline.py
CHANGED
@@ -8,14 +8,14 @@ import torch.nn.functional as F
|
|
8 |
from packaging import version
|
9 |
from transformers import CLIPImageProcessor, CLIPTextModel, CLIPTokenizer, CLIPVisionModelWithProjection
|
10 |
|
11 |
-
from diffusers.
|
12 |
-
from diffusers.
|
13 |
-
from diffusers.
|
14 |
-
from diffusers.
|
15 |
-
from diffusers.
|
16 |
-
from diffusers.
|
17 |
-
from diffusers.
|
18 |
-
from diffusers.
|
19 |
USE_PEFT_BACKEND,
|
20 |
deprecate,
|
21 |
logging,
|
@@ -23,12 +23,12 @@ from diffusers.src.diffusers.utils import (
|
|
23 |
scale_lora_layers,
|
24 |
unscale_lora_layers,
|
25 |
)
|
26 |
-
from diffusers.
|
27 |
-
from diffusers.
|
28 |
-
from diffusers.
|
29 |
-
from diffusers.
|
30 |
|
31 |
-
from diffusers.
|
32 |
|
33 |
|
34 |
logger = logging.get_logger(__name__) # pylint: disable=invalid-name
|
|
|
8 |
from packaging import version
|
9 |
from transformers import CLIPImageProcessor, CLIPTextModel, CLIPTokenizer, CLIPVisionModelWithProjection
|
10 |
|
11 |
+
from diffusers.configuration_utils import FrozenDict
|
12 |
+
from diffusers.image_processor import PipelineImageInput, VaeImageProcessor
|
13 |
+
from diffusers.loaders import FromSingleFileMixin, IPAdapterMixin, LoraLoaderMixin, TextualInversionLoaderMixin
|
14 |
+
from diffusers.models import AutoencoderKL, ImageProjection, UNet2DConditionModel
|
15 |
+
from diffusers.models.attention_processor import FusedAttnProcessor2_0
|
16 |
+
from diffusers.models.lora import adjust_lora_scale_text_encoder
|
17 |
+
from diffusers.schedulers import KarrasDiffusionSchedulers
|
18 |
+
from diffusers.utils import (
|
19 |
USE_PEFT_BACKEND,
|
20 |
deprecate,
|
21 |
logging,
|
|
|
23 |
scale_lora_layers,
|
24 |
unscale_lora_layers,
|
25 |
)
|
26 |
+
from diffusers.utils.torch_utils import randn_tensor
|
27 |
+
from diffusers.pipelines.pipeline_utils import DiffusionPipeline
|
28 |
+
from diffusers.pipelines.stable_diffusion.pipeline_output import StableDiffusionPipelineOutput
|
29 |
+
from diffusers.pipelines.stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
30 |
|
31 |
+
from diffusers.models.attention_processor import Attention, AttnProcessor2_0
|
32 |
|
33 |
|
34 |
logger = logging.get_logger(__name__) # pylint: disable=invalid-name
|