talatapp's picture
Fix cache orientation: keep NeMo native [L, B, ...] throughout instead of transposing to batch-major (#1290)
afa6bd0 verified
raw
history blame contribute delete
677 Bytes
{
"model_id": "nvidia/parakeet_realtime_eou_120m-v1",
"chunk_ms": 320,
"cursor_shift_samples": 5120,
"chunk_samples": 10080,
"encoder_window_mel_frames": 64,
"pre_cache_mel_frames": 9,
"sample_rate": 16000,
"vocab_size": 1026,
"vocab_with_blank": 1027,
"duration_bins": [],
"encoder_subsampling": 8,
"samples_per_encoder_frame": 1280,
"encoder_d_model": 512,
"encoder_layers": 17,
"encoder_cache_size": 70,
"encoder_conv_context": 4,
"decoder_hidden": 640,
"decoder_layers": 1,
"cache_last_channel_shape": [
17,
1,
70,
512
],
"cache_last_time_shape": [
17,
1,
512,
8
]
}