File size: 747 Bytes
c0c5723 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
{
"architectures": [
"SpeechTextAny2Any"
],
"bos_token_id": 1,
"complementary_masking": true,
"encoder_config": {
"attn_dropout": 0.0,
"attn_flash": true,
"conv_dropout": 0.0,
"conv_expansion_factor": 2,
"conv_kernel_size": 5,
"depth": 8,
"dim_head": 32,
"ff_dropout": 0.0,
"ff_mult": 4,
"heads": 16
},
"eos_token_id": 2,
"hidden_size": 512,
"label_smoothing": 0.0,
"model_type": "speech_text_any2any",
"num_speech_tokens": 1024,
"num_text_tokens": 256,
"pad_token_id": 0,
"special_tokens": {
"[MASK]": 3,
"[PAD]": 0,
"[SEP]": 4,
"[SPEECH]": 2,
"[TEXT]": 1
},
"torch_dtype": "bfloat16",
"transformers_version": "4.41.1",
"uniform_masking": true
}
|