&VjirUddlZddlmZmZddlmZGddZejaejej ej efe d<dej deddfd Z dej deefd Z dS) N)castOptionalceZdZdS)_StateN)__name__ __module__ __qualname__]/root/voice-cloning/.venv/lib/python3.11/site-packages/torch/distributed/_composable_state.pyrrsDr r_module_state_mappingmodulestatereturncl|tvsJd|dtj|t|<dS)Nz Inserting z more than once.)r weakrefref)rrs r _insert_module_statersC . . . .0UV0U0U0U . . .$+K$6$6&!!!r ct|trtt|S|tvr*t|}|}|t d|SdS)at Return the ``_State`` in ``model``. Given a ``module``, this API finds out if the module is also a ``_State`` instance or if the module is managed by a composable API. If the module is also a ``_State``, ``module`` will be casted to ``_State` and returned. If it is managed by a composable API, the corresponding ``_State`` will be returned. Nz(State has already been garbage collected) isinstancerrr AssertionError)r state_refrs r _get_module_statersg&&!! FF### * * *-f5IIKKE}$%OPPPL4r )rtypingrrtorch.nnnnrWeakKeyDictionaryr Module ReferenceType__annotations__rrr r r r!s!!!!!!!!        Gw0Iw$V,,   7767d7777 biHV,<r