o !if@sBUddlmZmZmZmZmZmZmZddlm Z m Z m Z ddl m Z ee e gefZiZeeeefed<dededdfd d Zd e d e deefd dZdeeefd e deefddZdeed e deefddZdeed e deefddZded e deefddZeeeeeeeeeee edS))CallableAnyTupleListDictType NamedTuple)PyTreeTreeSpecLeafSpec) namedtupleSUPPORTED_NODEStypflatten_fn_specreturnNcCs |t|<dS)N)r )rrr_/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/torch/fx/_pytree.pyregister_pytree_flatten_specs rpytreespeccCspt|tr|gS|jtvrtt|dt|j}|||}g}t||jD] \}}t||}||7}q(|S)Nz does not have a flatten_fn_spec associated with it. Please register one with torch.fx._pytree.register_pytree_flatten_spec. If you have serialized your model, make sure that any custom pytrees have been registered before loading it.) isinstancer typer RuntimeErrorzipchildren_specstree_flatten_spec)rrrZ child_pytreesresultchildZ child_specZflatrrrr s       rdcsfdd|jDS)Ncg|]}|qSrr).0krrr z&_dict_flatten_spec..)contextrrrr"r_dict_flatten_specsr'cfddtt|jDS)Ncrrrr ir"rrr#r$z&_list_flatten_spec..rangelenrr&rr"r_list_flatten_specr.cr()Ncrrrr)r"rrr#"r$z'_tuple_flatten_spec..r+r&rr"r_tuple_flatten_spec!r/r0cr()Ncrrrr)r"rrr#%r$z,_namedtuple_flatten_spec..r+r&rr"r_namedtuple_flatten_spec$r/r1)typingrrrrrrrZtorch.utils._pytreer r r collectionsr ZFlattenFuncSpecr __annotations__rrr'r.r0r1dictlisttuplerrrrs& "