diff --git a/config/examples/train_full_fine_tune_flex.yaml b/config/examples/train_full_fine_tune_flex.yaml
index e449bb8d..f386a3b1 100644
--- a/config/examples/train_full_fine_tune_flex.yaml
+++ b/config/examples/train_full_fine_tune_flex.yaml
@@ -74,6 +74,10 @@ config:
         # quantize: true
         # you can quantize just the T5 text encoder here to save vram
         quantize_te: true
+        # only train the transformer blocks
+        only_if_contains: 
+          - "transformer.transformer_blocks."
+          - "transformer.single_transformer_blocks."
       sample:
         sampler: "flowmatch" # must match train.noise_scheduler
         sample_every: 250 # sample every this many steps