pickStep method

ParakeetTdtStep pickStep({
  1. required List<double> tokenLogits,
  2. required List<double> durationLogits,
  3. required ParakeetTdtDecodeState state,
})

Implementation

ParakeetTdtStep pickStep({
  required List<double> tokenLogits,
  required List<double> durationLogits,
  required ParakeetTdtDecodeState state,
}) {
  final tokenId = _argmax(tokenLogits);
  final durationId = _argmax(durationLogits);
  return pickStepFromIds(
    tokenId: tokenId,
    durationId: durationId,
    state: state,
  );
}