Update pipeline.py

pipeline.py

@@ -79,7 +79,20 @@ 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=text_encoder_3,
+            tokenizer=tokenizer,
+            tokenizer_2=tokenizer_2,
+            tokenizer_3=tokenizer_3,
+            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
         )
@@ -252,7 +265,20 @@ 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=text_encoder_3,
+            tokenizer=tokenizer,
+            tokenizer_2=tokenizer_2,
+            tokenizer_3=tokenizer_3,
+            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
         )