From f37df472dfb325f1f72c90e2df56e72f40f850f0 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Thu, 27 Nov 2025 17:57:52 +0200 Subject: [PATCH] Kandinsky5 blocks for compile too --- nodes/model_optimization_nodes.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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):