+j SSKJr SSKrSSKrSSKJr SSKJrJ r J r \(a SSK J r SSKJ r SrSrS S jrSSS SS /4SS jjrg)) annotationsN) WrappersMP) TYPE_CHECKINGCallableOptional) ModelPatcher)WrapperExecutorz torch.compiletorch_compile_kwargsc^SU4SjjnU$)zC Create a wrapper that will refer to the compiled_diffusion_model. c>0nTR5H[upE[RRURU5X4'[RR URXE5 M] U"U0UD6UR5H/upE[RR URXE5 M1 $!WR5H/upE[RR URXE5 M1 f=f)N)itemscomfyutilsget_attr class_objset_attr)executorargskwargs orig_moduleskeyvaluecompiled_module_dicts E/home/wildlama/comfy/ComfyUI/comfy_api/torch_helpers/torch_compile.pyapply_torch_compile_wrapper@apply_torch_compile_factory..apply_torch_compile_wrappers EL288: $)KK$8$89K9KS$Q ! $$X%7%7D;T,V,*002  $$X%7%7D3l002  $$X%7%7D3s A8B??AD)rr )rrs` rapply_torch_compile_factoryrs E '&Fdiffusion_modelcXUR[R[5 U(dS/nUUUUUS.n 0n UH,n [R "SSUR U 50U D6X'M. [U S9n UR[R[U 5 XR['g)aH Perform torch.compile that will be applied at sample time for either the whole model or specific params of the BaseModel instance. When keys is None, it will default to using ["diffusion_model"], compiling the whole diffusion_model. When a list of keys is provided, it will perform torch.compile on only the selected modules. r )backendoptionsmode fullgraphdynamicmodel)rNr) remove_wrappers_with_keyr APPLY_MODEL COMPILE_KEYtorchcompileget_model_objectradd_wrapper_with_key model_optionsTORCH_COMPILE_KWARGS) r'r"r#r$r%r&keysrrcompile_kwargscompiled_modulesr wrapper_funcs rset_torch_compile_wrapperr5!s "":#9#9;G !" N % !,,S1! ! /-L z55{LQ0>,-r)rzdict[str, Callable]returnr) r'rr"strr#zOptional[dict[str, str]]r$z Optional[str]r&zOptional[bool]r1z list[str]) __future__rr+ comfy.utilsrcomfy.patcher_extensionrtypingrrrcomfy.model_patcherrr r*r0rr5rrrr=sd" .4407 -'"cg26%ae/@.A$?$1$?R`$?$-$?r