+h SSKJr SSKJr SSKJr SSKrSSKJr SSK rSSK J r J r SSK Jr SSKJrJr \R$"\5r"S S \R*5r\"S S \55r"S S\R*5r"SS\\ 5rg)) dataclass)pi)OptionalN) ConfigMixinregister_to_config) ModelMixin) BaseOutputloggingcn^\rSrSrSrS\4U4SjjrS\RS\R4Sjr Sr U=r $) StableAudioPositionalEmbeddingzUsed for continuous timedimc>[TU]5 US-S:XdeUS-n[R"[R "U55Ulg)Nr)super__init__nn Parametertorchrandnweights)selfrhalf_dim __class__s p/home/james-whalen/.local/lib/python3.13/site-packages/diffusers/pipelines/stable_audio/modeling_stable_audio.pyr'StableAudioPositionalEmbedding.__init__"s@ aA~~!8||EKK$9: timesreturncUSnXRS-S-[-n[R"UR 5UR 54SS9n[R"X4SS9nU$)N).Nr)r)rrrcatsincos)rrfreqs fouriereds rforward&StableAudioPositionalEmbedding.forward(sai  T**Q.3IIuyy{EIIK8bA IIu0b9 r)r) __name__ __module__ __qualname____firstlineno____doc__intrrTensorr(__static_attributes__ __classcell__rs@rr r s2";C; U\\ellrr c\rSrSr%SrSr\\R\ S'Sr \\R\ S'Sr \\R\ S'Sr g) StableAudioProjectionModelOutput0a Args: Class for StableAudio projection layer's outputs. text_hidden_states (`torch.Tensor` of shape `(batch_size, sequence_length, hidden_size)`, *optional*): Sequence of hidden-states obtained by linearly projecting the hidden-states for the text encoder. seconds_start_hidden_states (`torch.Tensor` of shape `(batch_size, 1, hidden_size)`, *optional*): Sequence of hidden-states obtained by linearly projecting the audio start hidden states. seconds_end_hidden_states (`torch.Tensor` of shape `(batch_size, 1, hidden_size)`, *optional*): Sequence of hidden-states obtained by linearly projecting the audio end hidden states. Ntext_hidden_statesseconds_start_hidden_statesseconds_end_hidden_states) r*r+r,r-r.r7rrr0__annotations__r8r9r1r:rrr5r50sE 26.5:>%,,!7>8<x 5