Update src/pipeline.py

src/pipeline.py

@@ -1,4 +1,4 @@
-from diffusers import FluxPipeline, AutoencoderKL, AutoencoderTiny
+# from diffusers import FluxPipeline, AutoencoderKL, AutoencoderTiny
 from diffusers.image_processor import VaeImageProcessor
 from diffusers.schedulers import FlowMatchEulerDiscreteScheduler
 from huggingface_hub.constants import HF_HUB_CACHE
@@ -43,7 +43,7 @@ def load_pipeline() -> Pipeline:
     
     pipeline = FluxPipeline.from_pretrained(
         ckpt_id,
-        vae=vae,
+        # vae=vae,
         revision=ckpt_revision,
         transformer=model,
         text_encoder_2=text_encoder_2,