Update pipeline.py

pipeline.py

@@ -3,6 +3,7 @@ import numpy as np
 from transformers import CLIPTextModel, CLIPTokenizer, T5EncoderModel, T5TokenizerFast, CLIPTextModelWithProjection
 from diffusers import FlowMatchEulerDiscreteScheduler, AutoPipelineForImage2Image, FluxPipeline, FluxTransformer2DModel
 from diffusers import StableDiffusion3Pipeline, AutoencoderKL, DiffusionPipeline
+from diffusers.image_processor import VaeImageProcessor
 from diffusers.loaders import FluxLoraLoaderMixin, FromSingleFileMixin, SD3LoraLoaderMixin
 from diffusers.utils import (
     USE_PEFT_BACKEND,
@@ -67,9 +68,6 @@ def prepare_timesteps(
 
 # FLUX pipeline function
 class FluxWithCFGPipeline(DiffusionPipeline, FluxLoraLoaderMixin, FromSingleFileMixin):
-    _optional_components = []
-    _callback_tensor_inputs = ["latents", "prompt_embeds"]
-    
     def __init__(
         self,
         scheduler: FlowMatchEulerDiscreteScheduler,