Skip to content

vllm_omni.model_executor.models.indextts2.gpt.embeddings ¶

Learned position embeddings for IndexTTS2 GPT model.

LearnedPositionEmbeddings ¶

Bases: Module

emb `instance-attribute` ¶

emb = nn.Embedding(seq_len, model_dim)

forward ¶

forward(x: Tensor) -> Tensor

get_fixed_embedding ¶

get_fixed_embedding(ind: int, dev: device) -> Tensor