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
|