Shankara-A-S's picture
Upload config.json with huggingface_hub
9eebfa8 verified
Raw
History Blame
13.9 kB
{
"architectures": [
"VoxtralRealtimeForConditionalGeneration"
],
"audio_config": {
"activation_function": "gelu",
"attention_dropout": 0.0,
"dtype": "bfloat16",
"head_dim": 64,
"hidden_act": "silu",
"hidden_size": 1280,
"initializer_range": 0.02,
"intermediate_size": 5120,
"max_position_embeddings": 1500,
"model_type": "voxtral_realtime_encoder",
"num_attention_heads": 32,
"num_hidden_layers": 32,
"num_mel_bins": 128,
"rms_norm_eps": 1e-05,
"rope_parameters": {
"rope_theta": 1000000.0,
"rope_type": "default"
},
"sliding_window": 750,
"vocab_size": 131072
},
"audio_length_per_tok": 8,
"default_num_delay_tokens": 6,
"downsample_factor": 4,
"dtype": "bfloat16",
"head_dim": 128,
"hidden_size": 3072,
"intermediate_size": 9216,
"max_position_embeddings": 131072,
"model_type": "voxtral_realtime",
"num_attention_heads": 32,
"num_hidden_layers": 26,
"num_key_value_heads": 8,
"projector_hidden_act": "gelu",
"quantization_config": {
"config_groups": {
"group_0": {
"format": "float-quantized",
"input_activations": {
"actorder": null,
"block_structure": null,
"dynamic": true,
"group_size": null,
"num_bits": 8,
"observer": null,
"observer_kwargs": {},
"scale_dtype": null,
"strategy": "token",
"symmetric": true,
"type": "float",
"zp_dtype": null
},
"output_activations": null,
"targets": [
"re:^whisper_encoder\\.whisper_encoder\\.layers\\.\\d+\\.layers\\.(self_attn\\.(q_proj|k_proj|v_proj|out_proj|qkv_proj)|mlp\\.(fc1|fc2|fc3|gate_up_proj|down_proj))$"
],
"weights": {
"actorder": null,
"block_structure": null,
"dynamic": false,
"group_size": null,
"num_bits": 8,
"observer": "memoryless_minmax",
"observer_kwargs": {},
"scale_dtype": null,
"strategy": "channel",
"symmetric": true,
"type": "float",
"zp_dtype": null
}
},
"group_1": {
"format": "pack-quantized",
"input_activations": null,
"output_activations": null,
"targets": [
"language_model.model.layers.0.self_attn.q_proj",
"language_model.model.layers.0.self_attn.k_proj",
"language_model.model.layers.0.self_attn.v_proj",
"language_model.model.layers.0.self_attn.o_proj",
"language_model.model.layers.0.mlp.gate_proj",
"language_model.model.layers.0.mlp.up_proj",
"language_model.model.layers.0.mlp.down_proj",
"language_model.model.layers.1.self_attn.q_proj",
"language_model.model.layers.1.self_attn.k_proj",
"language_model.model.layers.1.self_attn.v_proj",
"language_model.model.layers.1.self_attn.o_proj",
"language_model.model.layers.1.mlp.gate_proj",
"language_model.model.layers.1.mlp.up_proj",
"language_model.model.layers.1.mlp.down_proj",
"language_model.model.layers.2.self_attn.q_proj",
"language_model.model.layers.2.self_attn.k_proj",
"language_model.model.layers.2.self_attn.v_proj",
"language_model.model.layers.2.self_attn.o_proj",
"language_model.model.layers.2.mlp.gate_proj",
"language_model.model.layers.2.mlp.up_proj",
"language_model.model.layers.2.mlp.down_proj",
"language_model.model.layers.3.self_attn.q_proj",
"language_model.model.layers.3.self_attn.k_proj",
"language_model.model.layers.3.self_attn.v_proj",
"language_model.model.layers.3.self_attn.o_proj",
"language_model.model.layers.3.mlp.gate_proj",
"language_model.model.layers.3.mlp.up_proj",
"language_model.model.layers.3.mlp.down_proj",
"language_model.model.layers.4.self_attn.q_proj",
"language_model.model.layers.4.self_attn.k_proj",
"language_model.model.layers.4.self_attn.v_proj",
"language_model.model.layers.4.self_attn.o_proj",
"language_model.model.layers.4.mlp.gate_proj",
"language_model.model.layers.4.mlp.up_proj",
"language_model.model.layers.4.mlp.down_proj",
"language_model.model.layers.5.self_attn.q_proj",
"language_model.model.layers.5.self_attn.k_proj",
"language_model.model.layers.5.self_attn.v_proj",
"language_model.model.layers.5.self_attn.o_proj",
"language_model.model.layers.5.mlp.gate_proj",
"language_model.model.layers.5.mlp.up_proj",
"language_model.model.layers.5.mlp.down_proj",
"language_model.model.layers.6.self_attn.q_proj",
"language_model.model.layers.6.self_attn.k_proj",
"language_model.model.layers.6.self_attn.v_proj",
"language_model.model.layers.6.self_attn.o_proj",
"language_model.model.layers.6.mlp.gate_proj",
"language_model.model.layers.6.mlp.up_proj",
"language_model.model.layers.6.mlp.down_proj",
"language_model.model.layers.7.self_attn.q_proj",
"language_model.model.layers.7.self_attn.k_proj",
"language_model.model.layers.7.self_attn.v_proj",
"language_model.model.layers.7.self_attn.o_proj",
"language_model.model.layers.7.mlp.gate_proj",
"language_model.model.layers.7.mlp.up_proj",
"language_model.model.layers.7.mlp.down_proj",
"language_model.model.layers.8.self_attn.q_proj",
"language_model.model.layers.8.self_attn.k_proj",
"language_model.model.layers.8.self_attn.v_proj",
"language_model.model.layers.8.self_attn.o_proj",
"language_model.model.layers.8.mlp.gate_proj",
"language_model.model.layers.8.mlp.up_proj",
"language_model.model.layers.8.mlp.down_proj",
"language_model.model.layers.9.self_attn.q_proj",
"language_model.model.layers.9.self_attn.k_proj",
"language_model.model.layers.9.self_attn.v_proj",
"language_model.model.layers.9.self_attn.o_proj",
"language_model.model.layers.9.mlp.gate_proj",
"language_model.model.layers.9.mlp.up_proj",
"language_model.model.layers.9.mlp.down_proj",
"language_model.model.layers.10.self_attn.q_proj",
"language_model.model.layers.10.self_attn.k_proj",
"language_model.model.layers.10.self_attn.v_proj",
"language_model.model.layers.10.self_attn.o_proj",
"language_model.model.layers.10.mlp.gate_proj",
"language_model.model.layers.10.mlp.up_proj",
"language_model.model.layers.10.mlp.down_proj",
"language_model.model.layers.11.self_attn.q_proj",
"language_model.model.layers.11.self_attn.k_proj",
"language_model.model.layers.11.self_attn.v_proj",
"language_model.model.layers.11.self_attn.o_proj",
"language_model.model.layers.11.mlp.gate_proj",
"language_model.model.layers.11.mlp.up_proj",
"language_model.model.layers.11.mlp.down_proj",
"language_model.model.layers.12.self_attn.q_proj",
"language_model.model.layers.12.self_attn.k_proj",
"language_model.model.layers.12.self_attn.v_proj",
"language_model.model.layers.12.self_attn.o_proj",
"language_model.model.layers.12.mlp.gate_proj",
"language_model.model.layers.12.mlp.up_proj",
"language_model.model.layers.12.mlp.down_proj",
"language_model.model.layers.13.self_attn.q_proj",
"language_model.model.layers.13.self_attn.k_proj",
"language_model.model.layers.13.self_attn.v_proj",
"language_model.model.layers.13.self_attn.o_proj",
"language_model.model.layers.13.mlp.gate_proj",
"language_model.model.layers.13.mlp.up_proj",
"language_model.model.layers.13.mlp.down_proj",
"language_model.model.layers.14.self_attn.q_proj",
"language_model.model.layers.14.self_attn.k_proj",
"language_model.model.layers.14.self_attn.v_proj",
"language_model.model.layers.14.self_attn.o_proj",
"language_model.model.layers.14.mlp.gate_proj",
"language_model.model.layers.14.mlp.up_proj",
"language_model.model.layers.14.mlp.down_proj",
"language_model.model.layers.15.self_attn.q_proj",
"language_model.model.layers.15.self_attn.k_proj",
"language_model.model.layers.15.self_attn.v_proj",
"language_model.model.layers.15.self_attn.o_proj",
"language_model.model.layers.15.mlp.gate_proj",
"language_model.model.layers.15.mlp.up_proj",
"language_model.model.layers.15.mlp.down_proj",
"language_model.model.layers.16.self_attn.q_proj",
"language_model.model.layers.16.self_attn.k_proj",
"language_model.model.layers.16.self_attn.v_proj",
"language_model.model.layers.16.self_attn.o_proj",
"language_model.model.layers.16.mlp.gate_proj",
"language_model.model.layers.16.mlp.up_proj",
"language_model.model.layers.16.mlp.down_proj",
"language_model.model.layers.17.self_attn.q_proj",
"language_model.model.layers.17.self_attn.k_proj",
"language_model.model.layers.17.self_attn.v_proj",
"language_model.model.layers.17.self_attn.o_proj",
"language_model.model.layers.17.mlp.gate_proj",
"language_model.model.layers.17.mlp.up_proj",
"language_model.model.layers.17.mlp.down_proj",
"language_model.model.layers.18.self_attn.q_proj",
"language_model.model.layers.18.self_attn.k_proj",
"language_model.model.layers.18.self_attn.v_proj",
"language_model.model.layers.18.self_attn.o_proj",
"language_model.model.layers.18.mlp.gate_proj",
"language_model.model.layers.18.mlp.up_proj",
"language_model.model.layers.18.mlp.down_proj",
"language_model.model.layers.19.self_attn.q_proj",
"language_model.model.layers.19.self_attn.k_proj",
"language_model.model.layers.19.self_attn.v_proj",
"language_model.model.layers.19.self_attn.o_proj",
"language_model.model.layers.19.mlp.gate_proj",
"language_model.model.layers.19.mlp.up_proj",
"language_model.model.layers.19.mlp.down_proj",
"language_model.model.layers.20.self_attn.q_proj",
"language_model.model.layers.20.self_attn.k_proj",
"language_model.model.layers.20.self_attn.v_proj",
"language_model.model.layers.20.self_attn.o_proj",
"language_model.model.layers.20.mlp.gate_proj",
"language_model.model.layers.20.mlp.up_proj",
"language_model.model.layers.20.mlp.down_proj",
"language_model.model.layers.21.self_attn.q_proj",
"language_model.model.layers.21.self_attn.k_proj",
"language_model.model.layers.21.self_attn.v_proj",
"language_model.model.layers.21.self_attn.o_proj",
"language_model.model.layers.21.mlp.gate_proj",
"language_model.model.layers.21.mlp.up_proj",
"language_model.model.layers.21.mlp.down_proj",
"language_model.model.layers.22.self_attn.q_proj",
"language_model.model.layers.22.self_attn.k_proj",
"language_model.model.layers.22.self_attn.v_proj",
"language_model.model.layers.22.self_attn.o_proj",
"language_model.model.layers.22.mlp.gate_proj",
"language_model.model.layers.22.mlp.up_proj",
"language_model.model.layers.22.mlp.down_proj",
"language_model.model.layers.23.self_attn.q_proj",
"language_model.model.layers.23.self_attn.k_proj",
"language_model.model.layers.23.self_attn.v_proj",
"language_model.model.layers.23.self_attn.o_proj",
"language_model.model.layers.23.mlp.gate_proj",
"language_model.model.layers.23.mlp.up_proj",
"language_model.model.layers.23.mlp.down_proj",
"language_model.model.layers.24.self_attn.q_proj",
"language_model.model.layers.24.self_attn.k_proj",
"language_model.model.layers.24.self_attn.v_proj",
"language_model.model.layers.24.self_attn.o_proj",
"language_model.model.layers.24.mlp.gate_proj",
"language_model.model.layers.24.mlp.up_proj",
"language_model.model.layers.24.mlp.down_proj",
"language_model.model.layers.25.self_attn.q_proj",
"language_model.model.layers.25.self_attn.k_proj",
"language_model.model.layers.25.self_attn.v_proj",
"language_model.model.layers.25.self_attn.o_proj",
"language_model.model.layers.25.mlp.gate_proj",
"language_model.model.layers.25.mlp.up_proj",
"language_model.model.layers.25.mlp.down_proj"
],
"weights": {
"actorder": "dynamic",
"block_structure": null,
"dynamic": false,
"group_size": 128,
"num_bits": 4,
"observer": "memoryless_minmax",
"observer_kwargs": {},
"scale_dtype": null,
"strategy": "group",
"symmetric": true,
"type": "int",
"zp_dtype": null
}
}
},
"format": "mixed-precision",
"global_compression_ratio": null,
"ignore": [
"re:^audio_language_adapter(\\.|$)",
"re:^language_model\\.model\\.embed_tokens$",
"re:^language_model\\.lm_head$",
"re:^language_model\\.model\\.norm$",
"re:^language_model\\.model\\.layers\\.\\d+\\.ada_rms_norm_t_cond(\\.|$)",
"re:^language_model\\.model\\.layers\\.\\d+\\.(input_layernorm|post_attention_layernorm)$",
"re:^mm_streams_embeddings(\\.|$)",
"re:^layers\\.\\d+\\.ada_rms_norm_t_cond(\\.|$)",
"re:^layers\\.\\d+\\.(attention_norm|ffn_norm)$",
"re:^norm$"
],
"kv_cache_scheme": {
"num_bits": 8,
"type": "float",
"symmetric": true,
"strategy": "tensor",
"dynamic": false
},
"quant_method": "compressed-tensors",
"quantization_status": "compressed",
"sparsity_config": {},
"transform_config": {},
"version": "0.14.0.1"
},
"rms_norm_eps": 1e-05,
"text_config": {
"attention_dropout": 0.0,
"bos_token_id": 1,
"dtype": "bfloat16",
"eos_token_id": 2,
"head_dim": 128,
"hidden_act": "silu",
"hidden_size": 3072,
"initializer_range": 0.02,
"intermediate_size": 9216,
"max_position_embeddings": 131072,
"model_type": "voxtral_realtime_text",
"num_attention_heads": 32,
"num_hidden_layers": 26,
"num_key_value_heads": 8,
"pad_token_id": null,
"rms_norm_eps": 1e-05,
"rope_parameters": {
"rope_theta": 1000000.0,
"rope_type": "default"
},
"sliding_window": 8192,
"tie_word_embeddings": true,
"use_cache": true,
"vocab_size": 131072
},
"transformers_version": "5.5.4",
"vocab_size": 131072
}