xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2026-01-23 20:54:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

Kandinsky5 blocks for compile too

Browse Source
This commit is contained in:
kijai 2025-11-27 17:57:52 +02:00
parent 390d05fe7e
commit f37df472df
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nodes/model_optimization_nodes.py
View File

@ -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):