parakeet-rs 0.4.3

Fast ASR & Speaker Diarization with NVIDIA Parakeet via ONNX
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46

// Generated by scripts/generate_burn.py from decoder_joint.onnx with burn-onnx 0.22.0. Do not edit.
// Where each burn parameter's data comes from; see src/burn_backend/onnx.rs.
use crate::burn_backend::onnx::{Source::*, Transform::*, Weight};

pub(crate) const WEIGHTS: &[Weight] = &[
    Weight { path: "constant1", shape: Some(&[1025, 640]), source: Param("decoder.prediction.embed.weight", Copy) },
    Weight { path: "constant14", shape: Some(&[1]), source: Param("val_2", Copy) },
    Weight { path: "linear1.bias", shape: Some(&[640]), source: Param("joint_enc.bias", Copy) },
    Weight { path: "linear1.weight", shape: Some(&[1024, 640]), source: Param("val_128", Copy) },
    Weight { path: "linear2.bias", shape: Some(&[640]), source: Param("joint_pred.bias", Copy) },
    Weight { path: "linear2.weight", shape: Some(&[640, 640]), source: Param("val_130", Copy) },
    Weight { path: "linear3.bias", shape: Some(&[1025]), source: Param("joint_net.2.bias", Copy) },
    Weight { path: "linear3.weight", shape: Some(&[640, 1025]), source: Param("val_132", Copy) },
    Weight { path: "lstm1.cell_gate.hidden_transform.bias", shape: Some(&[640]), source: Zeros },
    Weight { path: "lstm1.cell_gate.hidden_transform.weight", shape: Some(&[640, 640]), source: Param("val_29", RowsT(1920, 2560)) },
    Weight { path: "lstm1.cell_gate.input_transform.bias", shape: Some(&[640]), source: Param("val_51", BiasSum(1920, 2560, 4480, 5120)) },
    Weight { path: "lstm1.cell_gate.input_transform.weight", shape: Some(&[640, 640]), source: Param("val_28", RowsT(1920, 2560)) },
    Weight { path: "lstm1.forget_gate.hidden_transform.bias", shape: Some(&[640]), source: Zeros },
    Weight { path: "lstm1.forget_gate.hidden_transform.weight", shape: Some(&[640, 640]), source: Param("val_29", RowsT(1280, 1920)) },
    Weight { path: "lstm1.forget_gate.input_transform.bias", shape: Some(&[640]), source: Param("val_51", BiasSum(1280, 1920, 3840, 4480)) },
    Weight { path: "lstm1.forget_gate.input_transform.weight", shape: Some(&[640, 640]), source: Param("val_28", RowsT(1280, 1920)) },
    Weight { path: "lstm1.input_gate.hidden_transform.bias", shape: Some(&[640]), source: Zeros },
    Weight { path: "lstm1.input_gate.hidden_transform.weight", shape: Some(&[640, 640]), source: Param("val_29", RowsT(0, 640)) },
    Weight { path: "lstm1.input_gate.input_transform.bias", shape: Some(&[640]), source: Param("val_51", BiasSum(0, 640, 2560, 3200)) },
    Weight { path: "lstm1.input_gate.input_transform.weight", shape: Some(&[640, 640]), source: Param("val_28", RowsT(0, 640)) },
    Weight { path: "lstm1.output_gate.hidden_transform.bias", shape: Some(&[640]), source: Zeros },
    Weight { path: "lstm1.output_gate.hidden_transform.weight", shape: Some(&[640, 640]), source: Param("val_29", RowsT(640, 1280)) },
    Weight { path: "lstm1.output_gate.input_transform.bias", shape: Some(&[640]), source: Param("val_51", BiasSum(640, 1280, 3200, 3840)) },
    Weight { path: "lstm1.output_gate.input_transform.weight", shape: Some(&[640, 640]), source: Param("val_28", RowsT(640, 1280)) },
    Weight { path: "lstm2.cell_gate.hidden_transform.bias", shape: Some(&[640]), source: Zeros },
    Weight { path: "lstm2.cell_gate.hidden_transform.weight", shape: Some(&[640, 640]), source: Param("val_92", RowsT(1920, 2560)) },
    Weight { path: "lstm2.cell_gate.input_transform.bias", shape: Some(&[640]), source: Param("val_114", BiasSum(1920, 2560, 4480, 5120)) },
    Weight { path: "lstm2.cell_gate.input_transform.weight", shape: Some(&[640, 640]), source: Param("val_91", RowsT(1920, 2560)) },
    Weight { path: "lstm2.forget_gate.hidden_transform.bias", shape: Some(&[640]), source: Zeros },
    Weight { path: "lstm2.forget_gate.hidden_transform.weight", shape: Some(&[640, 640]), source: Param("val_92", RowsT(1280, 1920)) },
    Weight { path: "lstm2.forget_gate.input_transform.bias", shape: Some(&[640]), source: Param("val_114", BiasSum(1280, 1920, 3840, 4480)) },
    Weight { path: "lstm2.forget_gate.input_transform.weight", shape: Some(&[640, 640]), source: Param("val_91", RowsT(1280, 1920)) },
    Weight { path: "lstm2.input_gate.hidden_transform.bias", shape: Some(&[640]), source: Zeros },
    Weight { path: "lstm2.input_gate.hidden_transform.weight", shape: Some(&[640, 640]), source: Param("val_92", RowsT(0, 640)) },
    Weight { path: "lstm2.input_gate.input_transform.bias", shape: Some(&[640]), source: Param("val_114", BiasSum(0, 640, 2560, 3200)) },
    Weight { path: "lstm2.input_gate.input_transform.weight", shape: Some(&[640, 640]), source: Param("val_91", RowsT(0, 640)) },
    Weight { path: "lstm2.output_gate.hidden_transform.bias", shape: Some(&[640]), source: Zeros },
    Weight { path: "lstm2.output_gate.hidden_transform.weight", shape: Some(&[640, 640]), source: Param("val_92", RowsT(640, 1280)) },
    Weight { path: "lstm2.output_gate.input_transform.bias", shape: Some(&[640]), source: Param("val_114", BiasSum(640, 1280, 3200, 3840)) },
    Weight { path: "lstm2.output_gate.input_transform.weight", shape: Some(&[640, 640]), source: Param("val_91", RowsT(640, 1280)) },
];