Update pipeline.py

pipeline.py

@@ -1,7 +1,7 @@
 import torch
 import numpy as np
 from diffusers import FlowMatchEulerDiscreteScheduler
-from diffusers import FluxPipeline
+from diffusers import StableDiffusion3Pipeline
 from diffusers.pipelines.flux.pipeline_output import FluxPipelineOutput
 from typing import Any, Callable, Dict, List, Optional, Union
 from PIL import Image
@@ -55,14 +55,14 @@ def prepare_timesteps(
     return timesteps, num_inference_steps
 
 # FLUX pipeline function
-class FluxWithCFGPipeline(FluxPipeline):
+class FluxWithCFGPipeline(StableDiffusion3Pipeline):
     def __call__(
         self,
         prompt: Union[str, List[str]] = None,
         prompt_2: Optional[Union[str, List[str]]] = None,
         height: Optional[int] = None,
         width: Optional[int] = None,
-        negative_prompt: Optional[Union[str, List[str]]] = None,
+        negative_prompt: Union[str, List[str]] = None,
         negative_prompt_2: Optional[Union[str, List[str]]] = None,
         num_inference_steps: int = 4,
         timesteps: List[int] = None,