Skip to content

vllm.model_executor.models.fireredasr2

Classes:

FireRedASR2AudioInputs

Bases: TensorSchema

Dimensions: - b: Batch size - nmb: Number of mel bins - t: Time frames (M)

Source code in vllm/model_executor/models/fireredasr2.py 
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
class FireRedASR2AudioInputs(TensorSchema):
    """Dimensions:
    - b: Batch size
    - nmb: Number of mel bins
    - t: Time frames (M)
    """

    input_features: Annotated[
        list[torch.Tensor] | None,
        TensorShape("b", "nmb", "t"),
    ]
    speech_lengths: Annotated[
        list[torch.Tensor] | None,
        TensorShape("b"),
    ]
    fake_token_lengths: Annotated[
        list[torch.Tensor] | None,
        TensorShape("b"),
    ]