[^(h ddlmZddlmZddlZdeej dej fdZdej deej ddfdZd ej d eedefd Z y) )Iterable)OptionalN parametersreturncd}g}|D].}t||}|j|jd0tj|S)zFlatten an iterable of parameters into a single vector. Args: parameters (Iterable[Tensor]): an iterable of Tensors that are the parameters of a model. Returns: The parameters represented by a single vector N)_check_param_deviceappendviewtorchcat)r param_devicevecparams _/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/torch/nn/utils/convert_parameters.pyparameters_to_vectorrsML C#*5,?  5::b>" # 99S>rct|tjs!tdtj|d}d}|D]I}t ||}|j }||||zj|j|_||z }Ky)aCopy slices of a vector into an iterable of parameters. Args: vec (Tensor): a single vector representing the parameters of a model. parameters (Iterable[Tensor]): an iterable of Tensors that are the parameters of a model. z expected torch.Tensor, but got: Nr) isinstancer Tensor TypeErrortypenamer numelview_asdata)rrrpointerr num_params rvector_to_parametersrs c5<< (:5>>#;N:OPQQLG *5,? KKM 7Y#67??FKK  9 rrold_param_devicec&dtjjg}|.|jj|vr|j }|Sd}|Sd}|jj|vr|j |k7}n|dk7}|r t d|S)a"Check if the parameters are located on the same device. Currently, the conversion between model parameters and single vector form is not supported for multiple allocations, e.g. parameters in different GPUs/PrivateUse1s, or mixture of CPU/GPU/PrivateUse1. Args: param ([Tensor]): a Tensor of a parameter of a model old_param_device (int): the device where the first parameter of a model is allocated. Returns: old_param_device (int): report device for the first time cudarFzKFound two parameters on different devices, this is currently not supported.)r _C_get_privateuse1_backend_namedevicetype get_devicer)rrsupport_device_typeswarns rr r :s#EHH$J$J$LM"',,"3"37K"KE     RT   LL  !5 5##%)99D#r)D 3  r) collections.abcrtypingrr rrrintr rrr-sy$ Xell%; ,ell8NSW:!u||!x}!QT!r