diff --git a/nodes/model_optimization_nodes.py b/nodes/model_optimization_nodes.py
index 2141e74..e9954fa 100644
--- a/nodes/model_optimization_nodes.py
+++ b/nodes/model_optimization_nodes.py
@@ -887,7 +887,7 @@ class TorchCompileModelAdvanced:
 
         try:
             if compile_transformer_blocks_only:
-                layer_types = ["double_blocks", "single_blocks", "layers", "transformer_blocks", "blocks"]
+                layer_types = ["double_blocks", "single_blocks", "layers", "transformer_blocks", "blocks", "visual_transformer_blocks", "text_transformer_blocks"]
                 compile_key_list = []
                 for layer_name in layer_types:
                     if hasattr(diffusion_model, layer_name):