Reza2kn's picture
Upload PhaseB streaming native artifacts
1ec470b verified
Raw
History Blame Contribute Delete
3.17 kB
{
"source_nemo": "/Users/studio/.cache/huggingface/hub/models--Reza2kn--visualears-fastconformer-fa-depoisoned-phaseB/snapshots/483fee957970c5bb2988b4f5682b63284b805889/depoison_phaseB2_final_8.02wer.nemo",
"format": "coreml-fixed-chunk-streaming",
"streaming_contract": {
"chunk_size": 4,
"shift_size": 3,
"left_chunks": 64,
"feature_chunk_frames": 32,
"feature_shift_frames": [
17,
24
],
"valid_out_len": 3,
"cache_last_channel_shape": [
17,
1,
256,
512
],
"cache_last_time_shape": [
17,
1,
512,
4
],
"heads": [
"rnnt",
"ctc"
]
},
"artifacts": [
{
"name": "streaming_encoder_ctc",
"status": "ok",
"torchscript": "artifacts/phaseB-coreml-fixed-streaming-fp16-v6/streaming_encoder_ctc_coreml_source_ts.pt",
"coreml": "artifacts/phaseB-coreml-fixed-streaming-fp16-v6/streaming_encoder_ctc_streaming_fp16.mlpackage",
"input_shapes": [
[
1,
80,
32
],
[
17,
1,
256,
512
],
[
17,
1,
512,
4
],
[
1
]
],
"output_names": [
"encoded",
"ctc_logits",
"next_cache_last_channel",
"next_cache_last_time",
"next_cache_last_channel_len"
],
"output_shapes": [
[
1,
512,
2
],
[
1,
2,
1025
],
[
17,
1,
256,
512
],
[
17,
1,
512,
4
],
[
1
]
]
},
{
"name": "rnnt_predictor",
"status": "ok",
"torchscript": "artifacts/phaseB-coreml-fixed-streaming-fp16-v6/rnnt_predictor_coreml_source_ts.pt",
"coreml": "artifacts/phaseB-coreml-fixed-streaming-fp16-v6/rnnt_predictor_streaming_fp16.mlpackage",
"input_shapes": [
[
1,
1
],
[
1,
1,
640
],
[
1,
1,
640
]
],
"output_names": [
"prediction",
"next_state_h",
"next_state_c"
],
"output_shapes": [
[
1,
1,
640
],
[
1,
1,
640
],
[
1,
1,
640
]
]
},
{
"name": "rnnt_joint",
"status": "ok",
"torchscript": "artifacts/phaseB-coreml-fixed-streaming-fp16-v6/rnnt_joint_coreml_source_ts.pt",
"coreml": "artifacts/phaseB-coreml-fixed-streaming-fp16-v6/rnnt_joint_streaming_fp16.mlpackage",
"input_shapes": [
[
1,
512,
2
],
[
1,
1,
640
]
],
"output_names": [
"rnnt_log_probs"
],
"output_shapes": [
[
1,
2,
1,
1025
]
]
}
]
}