Update pipeline.py

pipeline.py

@@ -70,28 +70,16 @@ class FluxWithCFGPipeline(StableDiffusion3Pipeline):
 
     def __init__(
         self,
-        transformer: FluxTransformer2DModel,
-        scheduler: FlowMatchEulerDiscreteScheduler,
-        vae: AutoencoderKL,
-        text_encoder: CLIPTextModelWithProjection,
-        tokenizer: CLIPTokenizer,
-        tokenizer_2: T5TokenizerFast,
-        tokenizer_3: None,
-        text_encoder_2: T5EncoderModel,
-        text_encoder_3: None,
-    ):
-        super().__init__(
-            self.register_modules(
-            vae=vae,
-            text_encoder=text_encoder,
-            text_encoder_2=text_encoder_2,
-            text_encoder_3=None,
-            tokenizer=tokenizer,
-            tokenizer_2=tokenizer_2,
-            tokenizer_3=None,
-            transformer=transformer,
-            scheduler=scheduler)
-        )
+        transformer = FluxTransformer2DModel,
+        scheduler = FlowMatchEulerDiscreteScheduler,
+        vae = AutoencoderKL,
+        text_encoder = CLIPTextModelWithProjection,
+        tokenizer = CLIPTokenizer,
+        tokenizer_2 = T5TokenizerFast,
+        tokenizer_3 = None,
+        text_encoder_2 = T5EncoderModel,
+        text_encoder_3 = None,
+    )
         self.vae_scale_factor = (
             2 ** (len(self.vae.config.block_out_channels) - 1) if hasattr(self, "vae") and self.vae is not None else 16
         )
@@ -264,19 +252,7 @@ class FluxWithCFGPipeline(StableDiffusion3Pipeline):
         tokenizer_3: None,
         text_encoder_2: T5EncoderModel,
         text_encoder_3: None,
-    ):
-        super().__init__(
-            self.register_modules(
-            vae=vae,
-            text_encoder=text_encoder,
-            text_encoder_2=text_encoder_2,
-            text_encoder_3=None,
-            tokenizer=tokenizer,
-            tokenizer_2=tokenizer_2,
-            tokenizer_3=None,
-            transformer=transformer,
-            scheduler=scheduler)
-        )
+    )
         self.vae_scale_factor = (
             2 ** (len(self.vae.config.block_out_channels) - 1) if hasattr(self, "vae") and self.vae is not None else 16
         )