o iA@sUdZddlZddlZddlZGdddZeZddZddZej Z iZ iZ d d Z d d Zd dZddZeZejed<eZejed<ddZddZddZddZdS)zJIT-related state This module stores various pieces of Python-global state relating to the JIT. This is not intended to be imported directly; please the exposed functionalities in `torch.jit`. Nc@s(eZdZdZddZddZddZdS) EnabledProxyzyStores whether the JIT is enabled or not. This is just a wrapper for a bool, so that we get reference semantics cCs|dddd|_dS)NZ PYTORCH_JITTz> Using PyTorch JITz> PyTorch JIT DISABLED) parse_envenabledselfr_/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/torch/jit/_state.py__init__s zEnabledProxy.__init__cCsptj|}|dur |S|dvrdS|dvrdS|dkr&t|dS|dkr0t|dStd|d) N>true1yesT>0nofalseFZ1vZ0vzUnknown setting of z. Try using 0 or 1.)osenvirongetlowerprint ValueError)rnamedefaultZ true_messageZ false_messagevaluerrrrs   zEnabledProxy.parse_envcCs|jSN)rrrrr__bool__)szEnabledProxy.__bool__N)__name__ __module__ __qualname____doc__r rrrrrrrs  rcC dt_dS)NF_enabledrrrrrdisable0 r"cCr)NTr rrrrenable4r#r$cCs|t|<|t|<dSr)_script_classes_name_to_pyclassqualified_name) python_classZ script_classrrr_add_script_classCsr)cCs(t|dd}|durt|}t|dS)NZ_jit_override_qualname)getattr_get_python_classr%r)r(overriderrr_get_script_classHs  r-cCs t|dSr)r&rr'rrrr+Os r+cCsttdSr)r%clearr&rrrr_clear_class_stateSs r0_jit_caching_layer_jit_function_overload_cachingcCs"t|d}|rdd|DSdS)NcSsg|]}t|qSr) _python_cu find_function).0 qual_namerrr gsz1_try_get_jit_cached_overloads..)r2r)keyZ qual_namesrrr_try_get_jit_cached_overloadsds r9cCsdd|Dt|<dS)NcSsg|]}|jqSrr.)r5fnrrrr7msz+_set_jit_overload_cache..)r2)r8Z compiled_fnsrrr_set_jit_overload_cachelsr;cCs2t|dddur dSt|d}|rt|SdS)NZ __disable_jit_function_caching__FT)r*r1rr3r4)r8r6rrr_try_get_jit_cached_functionps   r<cCs t|tjjs J|jt|<dSr) isinstancetorchZjitZScriptFunctionr'r1)r8rrrr_set_jit_function_cachezsr?)rrweakrefr>rr!r"r$Z_CZCompilationUnitr3r%r&r)r-r+r0WeakKeyDictionaryr1__annotations__r2r9r;r<r?rrrrs*