Update pipeline.py

pipeline.py

@@ -7,7 +7,7 @@ import numpy as np
 import torch
 from diffusers import DPMSolverMultistepScheduler
 from diffusers import StableDiffusionPipeline, AutoencoderKL
-from diffusers import Transformer2DModel, ModelMixin, ConfigMixin
+from diffusers import Transformer2DModel, ModelMixin, ConfigMixin, SchedulerMixin
 from diffusers import UNet2DConditionModel
 from diffusers.configuration_utils import register_to_config
 from diffusers.models.attention import BasicTransformerBlock
@@ -96,7 +96,7 @@ FlexibleUnetConfigurations = {
 }
 
 
-class SqueezedDPMSolverMultistepScheduler(DPMSolverMultistepScheduler):
+class SqueezedDPMSolverMultistepScheduler(DPMSolverMultistepScheduler, SchedulerMixin):
     """
     This is a copy-paste from Diffuser's `DPMSolverMultistepScheduler`, with minor differences:
     * Defaults are modified to accommodate DeciDiffusion