o ॵi9@s ddlmZddZddZdS))MappingcCsjddl} t|ttfrt|dd|DSt|tr%dd|DSt||jr3|}| S|S)z Numpify nested torch tensors. NOTE: If the type of input tensors is dict-like(Mapping, dict, OrderedDict, etc.), the return type will be dict. Args: tensors: Nested torch tensors. Returns: The numpify tensors. rNcs|]}t|VqdSNtorch_nested_numpify.0tr Q/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/utils/tensor_utils.py z'torch_nested_numpify..cSi|] \}}|t|qSr rrkr r r r z(torch_nested_numpify..) torch isinstancelisttupletyperitemsTensorcpunumpy)tensorsrr r r r rs   rcCsbddl} t|ttfrt|dd|DSt|tr%dd|DSt||jr/|S|S)z Detach nested torch tensors. NOTE: If the type of input tensors is dict-like(Mapping, dict, OrderedDict, etc.), the return type will be dict. Args: tensors: Nested torch tensors. Returns: The detached tensors. rNcsrrtorch_nested_detachrr r r r .r z&torch_nested_detach..cSrr rrr r r r0rz'torch_nested_detach..) rrrrrrrrdetach)rrr r r rs   rN)collections.abcrrrr r r r s