| { | |
| "architectures": [ | |
| "SpeechTextAny2Any" | |
| ], | |
| "bos_token_id": 1, | |
| "complementary_masking": false, | |
| "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 | |
| } | |