Update pipeline.py

pipeline.py

@@ -67,9 +67,7 @@ def prepare_timesteps(
 
 # FLUX pipeline function
 class FluxWithCFGPipeline(StableDiffusion3Pipeline):
-    _optional_components = [transformer, scheduler, vae, text_encoder, text_encoder_2, text_encoder_3, tokenizer, tokenizer_2, tokenizer_3]
-    model_cpu_offload_seq = "text_encoder->text_encoder_2->text_encoder_3->transformer->vae"
-    _callback_tensor_inputs = ["latents", "prompt_embeds", "negative_prompt_embeds", "negative_pooled_prompt_embeds"]
+
     def __init__(
         self,
         transformer: FluxTransformer2DModel,
@@ -102,6 +100,10 @@ class FluxWithCFGPipeline(StableDiffusion3Pipeline):
             self.tokenizer.model_max_length if hasattr(self, "tokenizer") and self.tokenizer is not None else 77
         )
         self.default_sample_size = 64
+        
+_optional_components = [transformer, scheduler, vae, text_encoder, text_encoder_2, text_encoder_3, tokenizer, tokenizer_2, tokenizer_3]
+model_cpu_offload_seq = "text_encoder->text_encoder_2->text_encoder_3->transformer->vae"
+_callback_tensor_inputs = ["latents", "prompt_embeds", "negative_prompt_embeds", "negative_pooled_prompt_embeds"]
 
     def __call__(
         self,