Rename pipeline.py to custom_pipeline.py

pipeline.py → custom_pipeline.py

@@ -1,7 +1,7 @@
 import torch
 import numpy as np
 from diffusers import FlowMatchEulerDiscreteScheduler
-from diffusers import DiffusionPipeline
+from diffusers import FluxPipeline
 from diffusers.pipelines.flux.pipeline_output import FluxPipelineOutput
 from typing import Any, Callable, Dict, List, Optional, Union
 from PIL import Image
@@ -55,7 +55,7 @@ def prepare_timesteps(
     return timesteps, num_inference_steps
 
 # FLUX pipeline function
-class FluxWithCFGPipeline(DiffusionPipeline):
+class FluxWithCFGPipeline(FluxPipeline):
     
     @torch.inference_mode()
     def generate_images(