mirror of
https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git
synced 2025-12-24 20:24:34 +08:00
Merge 853daefee1e08af9904e074099d434e69d3020b5 into 62a862db37d77a9a2e7611f638f9ff151a24fdec
This commit is contained in:
commit
1b656468a2
@ -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")
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user