Uh~dZddlZddlmZddlmZmZmZmZddl Z ddl Z ddl Z ddl m Z ddlmZddlmZdd lmZdd lmZdd lmZmZdd lmZdd lmZmZddlmZej@e!Z"eGddeZ#eGddeZ$e jJjLdZ' d=dZ(dZ)Gdde jjTZ+Gdde jTZ,Gdde jTZ-Gdde jTZ.Gdd e jTZ/Gd!d"e jTZ0Gd#d$e jTZ1Gd%d&e jTZ2Gd'd(e jTZ3Gd)d*e jTZ4Gd+d,e jTZ5Gd-d.e jTZ6Gd/d0e jTZ7Gd1d2e jTZ8Gd3d4e jTZ9Gd5d6e jTZ:eGd7d8eZ;ed9:Gd;dzPyTorch VITS model.N) dataclass)AnyOptionalTupleUnion)nn)ACT2FN)is_deepspeed_zero3_enabled)is_fsdp_managed_module)_prepare_4d_attention_mask)BaseModelOutput ModelOutput)PreTrainedModel)auto_docstringlogging) VitsConfigceZdZUdZdZeejed<dZ eejed<dZ ee ejed<dZ ee ejed<dZ ee ejed<y)VitsModelOutputaC Describes the outputs for the VITS model, with potential hidden states and attentions. Args: waveform (`torch.FloatTensor` of shape `(batch_size, sequence_length)`): The final audio waveform predicted by the model. sequence_lengths (`torch.FloatTensor` of shape `(batch_size,)`): The length in samples of each element in the `waveform` batch. spectrogram (`torch.FloatTensor` of shape `(batch_size, sequence_length, num_bins)`): The log-mel spectrogram predicted at the output of the flow model. This spectrogram is passed to the Hi-Fi GAN decoder model to obtain the final audio waveform. hidden_states (`tuple(torch.FloatTensor)`, *optional*, returned when `output_hidden_states=True` is passed or when `config.output_hidden_states=True`): Tuple of `torch.FloatTensor` (one for the output of the embeddings, if the model has an embedding layer, + one for the output of each layer) of shape `(batch_size, sequence_length, hidden_size)`. Hidden-states of the model at the output of each layer plus the optional initial embedding outputs. attentions (`tuple(torch.FloatTensor)`, *optional*, returned when `output_attentions=True` is passed or when `config.output_attentions=True`): Tuple of `torch.FloatTensor` (one for each layer) of shape `(batch_size, num_heads, sequence_length, sequence_length)`. Attention weights after the attention softmax, used to compute the weighted average in the self-attention heads. Nwaveformsequence_lengths spectrogram hidden_states attentions)__name__ __module__ __qualname____doc__rrtorch FloatTensor__annotations__rrrrrx/var/www/catia.catastroantioquia-mas.com/valormas/lib/python3.12/site-packages/transformers/models/vits/modeling_vits.pyrr's0-1Hhu(()048hu00186:K% 1 123:8