Changed FromCkptMixin to FromSingleFileMixin to fix error
Browse files- pipeline.py +2 -2
pipeline.py
CHANGED
@@ -14,7 +14,7 @@ from tqdm.auto import tqdm
|
|
14 |
from diffusers import DiffusionPipeline
|
15 |
from diffusers.configuration_utils import FrozenDict
|
16 |
from diffusers.image_processor import VaeImageProcessor
|
17 |
-
from diffusers.loaders import
|
18 |
from diffusers.models import AutoencoderKL, UNet2DConditionModel
|
19 |
from diffusers.pipelines.stable_diffusion import StableDiffusionPipelineOutput, StableDiffusionSafetyChecker
|
20 |
from diffusers.schedulers import KarrasDiffusionSchedulers
|
@@ -413,7 +413,7 @@ def preprocess_mask(mask, batch_size, scale_factor=8):
|
|
413 |
|
414 |
|
415 |
class StableDiffusionLongPromptWeightingPipeline(
|
416 |
-
DiffusionPipeline, TextualInversionLoaderMixin, LoraLoaderMixin,
|
417 |
):
|
418 |
r"""
|
419 |
Pipeline for text-to-image generation using Stable Diffusion without tokens length limit, and support parsing
|
|
|
14 |
from diffusers import DiffusionPipeline
|
15 |
from diffusers.configuration_utils import FrozenDict
|
16 |
from diffusers.image_processor import VaeImageProcessor
|
17 |
+
from diffusers.loaders import FromSingleFileMixin, LoraLoaderMixin, TextualInversionLoaderMixin
|
18 |
from diffusers.models import AutoencoderKL, UNet2DConditionModel
|
19 |
from diffusers.pipelines.stable_diffusion import StableDiffusionPipelineOutput, StableDiffusionSafetyChecker
|
20 |
from diffusers.schedulers import KarrasDiffusionSchedulers
|
|
|
413 |
|
414 |
|
415 |
class StableDiffusionLongPromptWeightingPipeline(
|
416 |
+
DiffusionPipeline, TextualInversionLoaderMixin, LoraLoaderMixin, FromSingleFileMixin
|
417 |
):
|
418 |
r"""
|
419 |
Pipeline for text-to-image generation using Stable Diffusion without tokens length limit, and support parsing
|