test_speech_unit / config.json
AlexHung29629's picture
Upload model
853e052 verified
raw
history blame contribute delete
436 Bytes
{
"architectures": [
"SpeechUnitModel"
],
"auto_map": {
"AutoConfig": "configuration_speechunit.SpeechUnitConfig",
"AutoModelForCausalLM": "modeling_speechunit.SpeechUnitModel"
},
"base_model_id": "meta-llama/Llama-3.2-1B",
"initializer_range": 0.02,
"model_type": "speechunit",
"num_heads": 8,
"num_hidden_layers": 3,
"output_dim": 2048,
"torch_dtype": "float32",
"transformers_version": "4.47.1"
}