diff --git a/nodes/model_optimization_nodes.py b/nodes/model_optimization_nodes.py index 543da07..844b478 100644 --- a/nodes/model_optimization_nodes.py +++ b/nodes/model_optimization_nodes.py @@ -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")