accentik / config.json
TeraSpace's picture
Update Accentik to audio v3 with opt-in ё placement
daad4bf verified
Raw History Blame Contribute Delete
3.62 kB
{
"architectures": [
"AccentikModel"
],
"auto_map": {
"AutoConfig": "configuration_accentik.AccentikConfig",
"AutoModel": "modeling_accentik.AccentikModel"
},
"batch_token_budget": 8192,
"context_window": 512,
"dtype": "float32",
"encoder_config": {
"accentor_local_head": true,
"activation_function": "gelu",
"allow_embedding_resizing": true,
"architectures": [
"ModernBertForMaskedLM"
],
"attention_bias": false,
"attention_dropout": 0.0,
"attention_layer": "rope",
"attention_probs_dropout_prob": 0.0,
"attn_out_bias": false,
"attn_out_dropout_prob": 0.1,
"attn_qkv_bias": false,
"bert_layer": "prenorm",
"bos_token_id": 50281,
"chunk_size_feed_forward": 0,
"classifier_activation": "gelu",
"classifier_bias": false,
"classifier_dropout": 0.0,
"classifier_pooling": "cls",
"cls_token_id": 50281,
"compile_model": true,
"decoder_bias": true,
"deterministic_flash_attn": false,
"dtype": "float32",
"embed_dropout_prob": 0.0,
"embed_norm": true,
"embedding_dropout": 0.0,
"embedding_layer": "sans_pos",
"eos_token_id": 50282,
"final_norm": true,
"global_attn_every_n_layers": 3,
"head_pred_act": "gelu",
"hidden_act": "gelu",
"hidden_activation": "gelu",
"hidden_size": 384,
"id2label": {
"0": "LABEL_0",
"1": "LABEL_1"
},
"init_method": "full_megatron",
"initializer_cutoff_factor": 2.0,
"initializer_range": 0.02,
"intermediate_size": 576,
"is_encoder_decoder": false,
"label2id": {
"LABEL_0": 0,
"LABEL_1": 1
},
"layer_types": [
"full_attention",
"sliding_attention",
"sliding_attention",
"full_attention",
"sliding_attention",
"sliding_attention",
"full_attention",
"sliding_attention",
"sliding_attention",
"full_attention",
"sliding_attention",
"sliding_attention"
],
"local_attention": 256,
"local_attn_rotary_emb_base": 10000.0,
"loss_function": "fa_cross_entropy",
"loss_kwargs": {
"reduction": "mean"
},
"masked_prediction": true,
"max_position_embeddings": 8192,
"mlp_bias": false,
"mlp_dropout": 0.0,
"mlp_dropout_prob": 0.0,
"mlp_in_bias": false,
"mlp_layer": "glu",
"mlp_out_bias": false,
"model_type": "modernbert",
"norm_bias": false,
"norm_eps": 1e-05,
"norm_kwargs": {
"bias": false,
"eps": 1e-05
},
"normalization": "layernorm",
"num_attention_heads": 6,
"num_hidden_layers": 12,
"output_attentions": false,
"output_hidden_states": false,
"pad_token_id": 50283,
"padding": "unpadded",
"problem_type": null,
"repad_logits_with_grad": false,
"return_dict": true,
"rope_parameters": {
"full_attention": {
"rope_theta": 160000.0,
"rope_type": "default"
},
"sliding_attention": {
"rope_theta": 10000.0,
"rope_type": "default"
}
},
"rotary_emb_base": 160000.0,
"rotary_emb_dim": null,
"rotary_emb_interleaved": false,
"rotary_emb_scale_base": null,
"sep_token_id": 50282,
"skip_first_prenorm": true,
"sparse_pred_ignore_index": -100,
"sparse_prediction": false,
"tie_word_embeddings": true,
"transformers_version": "5.15.1",
"unpad_embeddings": true,
"vocab_size": 50368
},
"model_type": "accentik",
"release": "audio-v3",
"restore_yo_default": false,
"tie_word_embeddings": false,
"transformers_version": "5.15.1",
"yo_threshold": 0.8
}