Merge 853daefee1e08af9904e074099d434e69d3020b5 into 62a862db37d77a9a2e7611f638f9ff151a24fdec

This commit is contained in:
Urle Sistiana 2025-12-06 01:15:03 +00:00 committed by GitHub
commit 1b656468a2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

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")