| { | |
| "architectures": [ | |
| "USADModel" | |
| ], | |
| "attention_dropout_p": 0.1, | |
| "attention_type": "mhsa", | |
| "auto_map": { | |
| "AutoConfig": "configuration_usad.USADConfig", | |
| "AutoModel": "modeling_usad.USADModel" | |
| }, | |
| "conv_dropout_p": 0.1, | |
| "conv_expansion_factor": 2, | |
| "conv_kernel_size": 31, | |
| "conv_pos": true, | |
| "conv_pos_depth": 5, | |
| "conv_pos_groups": 16, | |
| "conv_pos_width": 95, | |
| "conv_subsample_channels": 64, | |
| "conv_subsample_rate": 2, | |
| "encoder_dim": 768, | |
| "feed_forward_dropout_p": 0.1, | |
| "feed_forward_expansion_factor": 4, | |
| "half_step_residual": true, | |
| "input_dim": 128, | |
| "input_dropout_p": 0.0, | |
| "mamba_bidirectional": false, | |
| "mamba_d_conv": 4, | |
| "mamba_d_state": 16, | |
| "mamba_expand": 2, | |
| "model_type": "usad", | |
| "num_attention_heads": 12, | |
| "num_layers": 12, | |
| "subsample_normalization": true, | |
| "torch_dtype": "float32", | |
| "transformer_style": true, | |
| "transformers_version": "4.52.4", | |
| "use_framewise_subsample": true, | |
| "use_patchwise_subsample": false | |
| } | |