[Model] Remove Text Encoder Output Layer Hardcoding in Flux2 (#5589)

Signed-off-by: Alex Brooks <albrooks@redhat.com>

tests/model_tests/diffusion/tiny_configs/Flux2KleinPipeline/text_encoder/config.json

@@ -13,31 +13,6 @@
   "initializer_range": 0.02,
   "intermediate_size": 64,
   "layer_types": [
-    "full_attention",
-    "full_attention",
-    "full_attention",
-    "full_attention",
-    "full_attention",
-    "full_attention",
-    "full_attention",
-    "full_attention",
-    "full_attention",
-    "full_attention",
-    "full_attention",
-    "full_attention",
-    "full_attention",
-    "full_attention",
-    "full_attention",
-    "full_attention",
-    "full_attention",
-    "full_attention",
-    "full_attention",
-    "full_attention",
-    "full_attention",
-    "full_attention",
-    "full_attention",
-    "full_attention",
-    "full_attention",
     "full_attention",
     "full_attention",
     "full_attention"
@@ -46,7 +21,7 @@
   "max_window_layers": 36,
   "model_type": "qwen3",
   "num_attention_heads": 2,
-  "num_hidden_layers": 28,
+  "num_hidden_layers": 3,
   "num_key_value_heads": 2,
   "rms_norm_eps": 1e-06,
   "rope_scaling": null,
@@ -56,5 +31,6 @@
   "transformers_version": "4.56.1",
   "use_cache": true,
   "use_sliding_window": false,
-  "vocab_size": 151936
+  "vocab_size": 151936,
+  "text_encoder_out_layers":[0, 1, 2]
 }