| { | |
| "architectures": [ | |
| "FLMAudioForCausalLM" | |
| ], | |
| "attention_bias": true, | |
| "attention_dropout": 0.0, | |
| "aud_channel": 8, | |
| "aud_depthgpt": { | |
| "bias": false, | |
| "dropout": 0.0, | |
| "n_embd": 1024, | |
| "n_head": 16, | |
| "n_layer": 6, | |
| "use_cmlp": true, | |
| "use_rmsnorm": true, | |
| "use_swiglu": true | |
| }, | |
| "aud_vocab_size": 2050, | |
| "auto_map": { | |
| "AutoConfig": "configuration_flmaudio.FLMAudioConfig", | |
| "AutoModel": "modeling_flmaudio.FLMAudioModel", | |
| "AutoModelForCausalLM": "modeling_flmaudio.FLMAudioForCausalLM" | |
| }, | |
| "bos_token_id": 151643, | |
| "disable_att_o_bias": true, | |
| "eos_token_id": 151645, | |
| "hidden_act": "silu", | |
| "hidden_size": 3584, | |
| "initializer_range": 0.02, | |
| "input_mult": 1.0, | |
| "intermediate_size": 18944, | |
| "max_position_embeddings": 8192, | |
| "mm_token_info": { | |
| "aud_emp_token_id": 2049, | |
| "aud_pad_token_id": 2048, | |
| "text_wait_token_id": 151665 | |
| }, | |
| "model_type": "FLMAudio", | |
| "mup_scale_factor": 28.0, | |
| "num_attention_heads": 28, | |
| "num_hidden_layers": 28, | |
| "num_key_value_heads": 4, | |
| "output_mult": 28.0, | |
| "pretraining_tp": 1, | |
| "rms_norm_eps": 1e-05, | |
| "rope_scaling": { | |
| "mrope_section": [ | |
| 16, | |
| 24, | |
| 24 | |
| ], | |
| "rope_type": "default", | |
| "type": "default" | |
| }, | |
| "rope_theta": 1000000, | |
| "sliding_window": 32768, | |
| "tie_word_embeddings": false, | |
| "torch_dtype": "bfloat16", | |
| "transformers_version": "4.53.1", | |
| "use_cache": true, | |
| "use_mup": true, | |
| "use_sliding_window": false, | |
| "vocab_size": 151668 | |
| } | |