{ "blank_token_id": 8192, "decoder_hidden_size": 640, "dtype": "float32", "durations": [ 0, 1, 2, 3, 4 ], "encoder_config": { "activation_dropout": 0.1, "attention_bias": false, "attention_dropout": 0.1, "conv_kernel_size": 9, "convolution_bias": false, "dropout": 0.1, "dropout_positions": 0.0, "hidden_act": "silu", "hidden_size": 1024, "initializer_range": 0.02, "intermediate_size": 4096, "layerdrop": 0.1, "max_position_embeddings": 5000, "model_type": "parakeet_encoder", "num_attention_heads": 8, "num_hidden_layers": 24, "num_key_value_heads": 8, "num_mel_bins": 128, "scale_input": false, "subsampling_conv_channels": 256, "subsampling_conv_kernel_size": 3, "subsampling_conv_stride": 2, "subsampling_factor": 8 }, "hidden_act": "relu", "initializer_range": 0.02, "is_encoder_decoder": true, "max_symbols_per_step": 10, "model_type": "parakeet_tdt", "num_decoder_layers": 2, "pad_token_id": 2, "vocab_size": 8193, "ternary_group_size": 128, "ternary_modules": [ "encoder.layers.0.feed_forward1.linear1", "encoder.layers.0.feed_forward1.linear2", "encoder.layers.0.self_attn.q_proj", "encoder.layers.0.self_attn.k_proj", "encoder.layers.0.self_attn.v_proj", "encoder.layers.0.self_attn.o_proj", "encoder.layers.0.self_attn.relative_k_proj", "encoder.layers.0.conv.pointwise_conv1", "encoder.layers.0.conv.pointwise_conv2", "encoder.layers.0.feed_forward2.linear1", "encoder.layers.0.feed_forward2.linear2", "encoder.layers.1.feed_forward1.linear1", "encoder.layers.1.feed_forward1.linear2", "encoder.layers.1.self_attn.q_proj", "encoder.layers.1.self_attn.k_proj", "encoder.layers.1.self_attn.v_proj", "encoder.layers.1.self_attn.o_proj", "encoder.layers.1.self_attn.relative_k_proj", "encoder.layers.1.conv.pointwise_conv1", "encoder.layers.1.conv.pointwise_conv2", "encoder.layers.1.feed_forward2.linear1", "encoder.layers.1.feed_forward2.linear2", "encoder.layers.2.feed_forward1.linear1", "encoder.layers.2.feed_forward1.linear2", "encoder.layers.2.self_attn.q_proj", "encoder.layers.2.self_attn.k_proj", "encoder.layers.2.self_attn.v_proj", "encoder.layers.2.self_attn.o_proj", "encoder.layers.2.self_attn.relative_k_proj", "encoder.layers.2.conv.pointwise_conv1", "encoder.layers.2.conv.pointwise_conv2", "encoder.layers.2.feed_forward2.linear1", "encoder.layers.2.feed_forward2.linear2", "encoder.layers.3.feed_forward1.linear1", "encoder.layers.3.feed_forward1.linear2", "encoder.layers.3.self_attn.q_proj", "encoder.layers.3.self_attn.k_proj", "encoder.layers.3.self_attn.v_proj", "encoder.layers.3.self_attn.o_proj", "encoder.layers.3.self_attn.relative_k_proj", "encoder.layers.3.conv.pointwise_conv1", "encoder.layers.3.conv.pointwise_conv2", "encoder.layers.3.feed_forward2.linear1", "encoder.layers.3.feed_forward2.linear2", "encoder.layers.4.feed_forward1.linear1", "encoder.layers.4.feed_forward1.linear2", "encoder.layers.4.self_attn.q_proj", "encoder.layers.4.self_attn.k_proj", "encoder.layers.4.self_attn.v_proj", "encoder.layers.4.self_attn.o_proj", "encoder.layers.4.self_attn.relative_k_proj", "encoder.layers.4.conv.pointwise_conv1", "encoder.layers.4.conv.pointwise_conv2", "encoder.layers.4.feed_forward2.linear1", "encoder.layers.4.feed_forward2.linear2", "encoder.layers.5.feed_forward1.linear1", "encoder.layers.5.feed_forward1.linear2", "encoder.layers.5.self_attn.q_proj", "encoder.layers.5.self_attn.k_proj", "encoder.layers.5.self_attn.v_proj", "encoder.layers.5.self_attn.o_proj", "encoder.layers.5.self_attn.relative_k_proj", "encoder.layers.5.conv.pointwise_conv1", "encoder.layers.5.conv.pointwise_conv2", "encoder.layers.5.feed_forward2.linear1", "encoder.layers.5.feed_forward2.linear2", "encoder.layers.6.feed_forward1.linear1", "encoder.layers.6.feed_forward1.linear2", "encoder.layers.6.self_attn.q_proj", "encoder.layers.6.self_attn.k_proj", "encoder.layers.6.self_attn.v_proj", "encoder.layers.6.self_attn.o_proj", "encoder.layers.6.self_attn.relative_k_proj", "encoder.layers.6.conv.pointwise_conv1", "encoder.layers.6.conv.pointwise_conv2", "encoder.layers.6.feed_forward2.linear1", "encoder.layers.6.feed_forward2.linear2", "encoder.layers.7.feed_forward1.linear1", "encoder.layers.7.feed_forward1.linear2", "encoder.layers.7.self_attn.q_proj", "encoder.layers.7.self_attn.k_proj", "encoder.layers.7.self_attn.v_proj", "encoder.layers.7.self_attn.o_proj", "encoder.layers.7.self_attn.relative_k_proj", "encoder.layers.7.conv.pointwise_conv1", "encoder.layers.7.conv.pointwise_conv2", "encoder.layers.7.feed_forward2.linear1", "encoder.layers.7.feed_forward2.linear2", "encoder.layers.8.feed_forward1.linear1", "encoder.layers.8.feed_forward1.linear2", "encoder.layers.8.self_attn.q_proj", "encoder.layers.8.self_attn.k_proj", "encoder.layers.8.self_attn.v_proj", "encoder.layers.8.self_attn.o_proj", "encoder.layers.8.self_attn.relative_k_proj", "encoder.layers.8.conv.pointwise_conv1", "encoder.layers.8.conv.pointwise_conv2", "encoder.layers.8.feed_forward2.linear1", "encoder.layers.8.feed_forward2.linear2", "encoder.layers.9.feed_forward1.linear1", "encoder.layers.9.feed_forward1.linear2", "encoder.layers.9.self_attn.q_proj", "encoder.layers.9.self_attn.k_proj", "encoder.layers.9.self_attn.v_proj", "encoder.layers.9.self_attn.o_proj", "encoder.layers.9.self_attn.relative_k_proj", "encoder.layers.9.conv.pointwise_conv1", "encoder.layers.9.conv.pointwise_conv2", "encoder.layers.9.feed_forward2.linear1", "encoder.layers.9.feed_forward2.linear2", "encoder.layers.10.feed_forward1.linear1", "encoder.layers.10.feed_forward1.linear2", "encoder.layers.10.self_attn.q_proj", "encoder.layers.10.self_attn.k_proj", "encoder.layers.10.self_attn.v_proj", "encoder.layers.10.self_attn.o_proj", "encoder.layers.10.self_attn.relative_k_proj", "encoder.layers.10.conv.pointwise_conv1", "encoder.layers.10.conv.pointwise_conv2", "encoder.layers.10.feed_forward2.linear1", "encoder.layers.10.feed_forward2.linear2", "encoder.layers.11.feed_forward1.linear1", "encoder.layers.11.feed_forward1.linear2", "encoder.layers.11.self_attn.q_proj", "encoder.layers.11.self_attn.k_proj", "encoder.layers.11.self_attn.v_proj", "encoder.layers.11.self_attn.o_proj", "encoder.layers.11.self_attn.relative_k_proj", "encoder.layers.11.conv.pointwise_conv1", "encoder.layers.11.conv.pointwise_conv2", "encoder.layers.11.feed_forward2.linear1", "encoder.layers.11.feed_forward2.linear2", "encoder.layers.12.feed_forward1.linear1", "encoder.layers.12.feed_forward1.linear2", "encoder.layers.12.self_attn.q_proj", "encoder.layers.12.self_attn.k_proj", "encoder.layers.12.self_attn.v_proj", "encoder.layers.12.self_attn.o_proj", "encoder.layers.12.self_attn.relative_k_proj", "encoder.layers.12.conv.pointwise_conv1", "encoder.layers.12.conv.pointwise_conv2", "encoder.layers.12.feed_forward2.linear1", "encoder.layers.12.feed_forward2.linear2", "encoder.layers.13.feed_forward1.linear1", "encoder.layers.13.feed_forward1.linear2", "encoder.layers.13.self_attn.q_proj", "encoder.layers.13.self_attn.k_proj", "encoder.layers.13.self_attn.v_proj", "encoder.layers.13.self_attn.o_proj", "encoder.layers.13.self_attn.relative_k_proj", "encoder.layers.13.conv.pointwise_conv1", "encoder.layers.13.conv.pointwise_conv2", "encoder.layers.13.feed_forward2.linear1", "encoder.layers.13.feed_forward2.linear2", "encoder.layers.14.feed_forward1.linear1", "encoder.layers.14.feed_forward1.linear2", "encoder.layers.14.self_attn.q_proj", "encoder.layers.14.self_attn.k_proj", "encoder.layers.14.self_attn.v_proj", "encoder.layers.14.self_attn.o_proj", "encoder.layers.14.self_attn.relative_k_proj", "encoder.layers.14.conv.pointwise_conv1", "encoder.layers.14.conv.pointwise_conv2", "encoder.layers.14.feed_forward2.linear1", "encoder.layers.14.feed_forward2.linear2", "encoder.layers.15.feed_forward1.linear1", "encoder.layers.15.feed_forward1.linear2", "encoder.layers.15.self_attn.q_proj", "encoder.layers.15.self_attn.k_proj", "encoder.layers.15.self_attn.v_proj", "encoder.layers.15.self_attn.o_proj", "encoder.layers.15.self_attn.relative_k_proj", "encoder.layers.15.conv.pointwise_conv1", "encoder.layers.15.conv.pointwise_conv2", "encoder.layers.15.feed_forward2.linear1", "encoder.layers.15.feed_forward2.linear2", "encoder.layers.16.feed_forward1.linear1", "encoder.layers.16.feed_forward1.linear2", "encoder.layers.16.self_attn.q_proj", "encoder.layers.16.self_attn.k_proj", "encoder.layers.16.self_attn.v_proj", "encoder.layers.16.self_attn.o_proj", "encoder.layers.16.self_attn.relative_k_proj", "encoder.layers.16.conv.pointwise_conv1", "encoder.layers.16.conv.pointwise_conv2", "encoder.layers.16.feed_forward2.linear1", "encoder.layers.16.feed_forward2.linear2", "encoder.layers.17.feed_forward1.linear1", "encoder.layers.17.feed_forward1.linear2", "encoder.layers.17.self_attn.q_proj", "encoder.layers.17.self_attn.k_proj", "encoder.layers.17.self_attn.v_proj", "encoder.layers.17.self_attn.o_proj", "encoder.layers.17.self_attn.relative_k_proj", "encoder.layers.17.conv.pointwise_conv1", "encoder.layers.17.conv.pointwise_conv2", "encoder.layers.17.feed_forward2.linear1", "encoder.layers.17.feed_forward2.linear2", "encoder.layers.18.feed_forward1.linear1", "encoder.layers.18.feed_forward1.linear2", "encoder.layers.18.self_attn.q_proj", "encoder.layers.18.self_attn.k_proj", "encoder.layers.18.self_attn.v_proj", "encoder.layers.18.self_attn.o_proj", "encoder.layers.18.self_attn.relative_k_proj", "encoder.layers.18.conv.pointwise_conv1", "encoder.layers.18.conv.pointwise_conv2", "encoder.layers.18.feed_forward2.linear1", "encoder.layers.18.feed_forward2.linear2", "encoder.layers.19.feed_forward1.linear1", "encoder.layers.19.feed_forward1.linear2", "encoder.layers.19.self_attn.q_proj", "encoder.layers.19.self_attn.k_proj", "encoder.layers.19.self_attn.v_proj", "encoder.layers.19.self_attn.o_proj", "encoder.layers.19.self_attn.relative_k_proj", "encoder.layers.19.conv.pointwise_conv1", "encoder.layers.19.conv.pointwise_conv2", "encoder.layers.19.feed_forward2.linear1", "encoder.layers.19.feed_forward2.linear2", "encoder.layers.20.feed_forward1.linear1", "encoder.layers.20.feed_forward1.linear2", "encoder.layers.20.self_attn.q_proj", "encoder.layers.20.self_attn.k_proj", "encoder.layers.20.self_attn.v_proj", "encoder.layers.20.self_attn.o_proj", "encoder.layers.20.self_attn.relative_k_proj", "encoder.layers.20.conv.pointwise_conv1", "encoder.layers.20.conv.pointwise_conv2", "encoder.layers.20.feed_forward2.linear1", "encoder.layers.20.feed_forward2.linear2", "encoder.layers.21.feed_forward1.linear1", "encoder.layers.21.feed_forward1.linear2", "encoder.layers.21.self_attn.q_proj", "encoder.layers.21.self_attn.k_proj", "encoder.layers.21.self_attn.v_proj", "encoder.layers.21.self_attn.o_proj", "encoder.layers.21.self_attn.relative_k_proj", "encoder.layers.21.conv.pointwise_conv1", "encoder.layers.21.conv.pointwise_conv2", "encoder.layers.21.feed_forward2.linear1", "encoder.layers.21.feed_forward2.linear2", "encoder.layers.22.feed_forward1.linear1", "encoder.layers.22.feed_forward1.linear2", "encoder.layers.22.self_attn.q_proj", "encoder.layers.22.self_attn.k_proj", "encoder.layers.22.self_attn.v_proj", "encoder.layers.22.self_attn.o_proj", "encoder.layers.22.self_attn.relative_k_proj", "encoder.layers.22.conv.pointwise_conv1", "encoder.layers.22.conv.pointwise_conv2", "encoder.layers.22.feed_forward2.linear1", "encoder.layers.22.feed_forward2.linear2", "encoder.layers.23.feed_forward1.linear1", "encoder.layers.23.feed_forward1.linear2", "encoder.layers.23.self_attn.q_proj", "encoder.layers.23.self_attn.k_proj", "encoder.layers.23.self_attn.v_proj", "encoder.layers.23.self_attn.o_proj", "encoder.layers.23.self_attn.relative_k_proj", "encoder.layers.23.conv.pointwise_conv1", "encoder.layers.23.conv.pointwise_conv2", "encoder.layers.23.feed_forward2.linear1", "encoder.layers.23.feed_forward2.linear2" ] }