Struct CodePredictorConfig 

pub struct CodePredictorConfig {
    pub hidden_size: usize,
    pub intermediate_size: usize,
    pub num_hidden_layers: usize,
    pub num_attention_heads: usize,
    pub num_key_value_heads: usize,
    pub head_dim: usize,
    pub rms_norm_eps: f64,
    pub rope_theta: f64,
    pub vocab_size: usize,
    pub num_code_groups: usize,
    pub codec_embed_dim: Option<usize>,
}

Code predictor configuration

Fields

hidden_size: usize

Hidden dimension

intermediate_size: usize

Intermediate size for MLP

num_hidden_layers: usize

Number of transformer layers

num_attention_heads: usize

Number of attention heads

num_key_value_heads: usize

Number of KV heads (for GQA)

head_dim: usize

Head dimension

rms_norm_eps: f64

RMS norm epsilon

rope_theta: f64

RoPE theta

vocab_size: usize

Vocabulary size for codec tokens

num_code_groups: usize

Number of code groups (total, including semantic)

codec_embed_dim: Option<usize>

Codec embedding dimension (may differ from hidden_size for CustomVoice models) When different from hidden_size, a small_to_mtp_projection is used

Implementations

impl CodePredictorConfig

pub fn from_parsed(parsed: &ParsedModelConfig) -> Self

Create config from parsed HuggingFace config.json.

When the talker hidden_size differs from the code predictor hidden_size (e.g. 1.7B models: talker=2048, CP=1024), codec_embed_dim is set to the talker’s hidden_size so the small_to_mtp_projection layer is created.

pub fn codec_embed_dim(&self) -> usize

Get the codec embedding dimension (defaults to hidden_size)

pub fn custom_voice() -> Self

Create config for CustomVoice model

Trait Implementations

impl Clone for CodePredictorConfig

fn clone(&self) -> CodePredictorConfig

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for CodePredictorConfig

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for CodePredictorConfig

fn default() -> Self

Returns the “default value” for a type.