AlanB commited on
Commit
89be630
·
1 Parent(s): bf344dc

Changed FromCkptMixin to FromSingleFileMixin to fix error

Browse files
Files changed (1) hide show
  1. 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 FromCkptMixin, 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,7 +413,7 @@ def preprocess_mask(mask, batch_size, scale_factor=8):
413
 
414
 
415
  class StableDiffusionLongPromptWeightingPipeline(
416
- DiffusionPipeline, TextualInversionLoaderMixin, LoraLoaderMixin, FromCkptMixin
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