Update pipeline.py

pipeline.py

@@ -37,7 +37,7 @@ from diffusers.utils import (
     unscale_lora_layers,
 )
 from diffusers.pipelines.stable_diffusion_3.pipeline_output import StableDiffusion3PipelineOutput
-from diffusers import DiffusionPipeline, ImagePipelineOutput
+from diffusers import StableDiffusion3Pipeline
 
 if is_torch_xla_available():
     import torch_xla.core.xla_model as xm
@@ -48,7 +48,7 @@ else:
 
 
 
-class SwDPipeline(DiffusionPipeline):
+class SwDPipeline(StableDiffusion3Pipeline):
 
     def __init__(
             self,