Update pipeline.py

pipeline.py

@@ -1,11 +1,11 @@
 import torch
 import numpy as np
-from diffusers import FlowMatchEulerDiscreteScheduler
-from diffusers import StableDiffusion3Pipeline
+from diffusers import FlowMatchEulerDiscreteScheduler, AutoPipelineForImage2Image, FluxPipeline, CLIPTokenizer, T5TokenizerFast, T5EncoderModel,
+from diffusers import StableDiffusion3Pipeline, AutoencoderKL, CLIPTextModelWithProjection
 from diffusers.pipelines.flux.pipeline_output import FluxPipelineOutput
 from typing import Any, Callable, Dict, List, Optional, Union
 from PIL import Image
-from diffusers.pipelines.flux.pipeline_flux import calculate_shift, retrieve_timesteps
+from diffusers.pipelines.flux.pipeline_flux import calculate_shift, retrieve_timesteps, FluxTransformer2DModel
 
 from diffusers.utils import is_torch_xla_available