xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2025-12-26 14:38:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

set default dynamic=None in TorchCompileModelAdvanced

Browse Source
This commit is contained in:
Urle Sistiana 2025-12-06 08:55:01 +08:00 committed by GitHub
parent 62a862db37
commit 853daefee1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nodes/model_optimization_nodes.py
View File

@ -904,6 +904,8 @@ class TorchCompileModelAdvanced:
if not compile_key_list:
compile_key_list =["diffusion_model"]
if not dynamic:
dynamic = None
set_torch_compile_wrapper(model=m, keys=compile_key_list, backend=backend, mode=mode, dynamic=dynamic, fullgraph=fullgraph)
except:
raise RuntimeError("Failed to compile model")