Skip to main content

← All model recipe tasks

canary model family

Tasks

Supported architectures and task heads

The Hugging Face values below are copied from checkpoint metadata at the recorded revision. The TRTMC task contract comes from the exact E2E recipe. Architecture identity selects or describes a source graph; it does not imply that TRTMC reproduces every Hugging Face head with the same model_type. For example, an encoder recipe may consume the base model and intentionally return hidden states instead of the checkpoint's pretraining or classification logits.

HF model_typeHF architecture / pipeline classTRTMC task contractExact recipe profiles
nemo.collections.asr.models.aed_multitask_models.EncDecMultiTaskModel
Source: model_config.target
Metadata: canary-1b-v2.nemo:model_config.yaml at 87bc52657add
speech_to_text
canary-1b-v2
canary-1b-v2-tp4

Declared recipes

Each row comes from a model-owned E2E manifest. It declares a test recipe; it is not a live pass receipt for every hardware target.

RecipeExact Hugging Face checkpointTaskBuild configurationDeclared E2E casesManifest
canary-1b-v2nvidia/canary-1b-v2fp16; single devicecanary-1b-v2, canary-1b-v2-asr-probe01, canary-1b-v2-asr-probe02, canary-1b-v2-asr-probe03, canary-1b-v2-asr-probe04, canary-1b-v2-asr-probe05, canary-1b-v2-asr-probe06, canary-1b-v2-asr-probe08tests/e2e/models/canary/manifests/canary-1b-v2.json
canary-1b-v2-tp4nvidia/canary-1b-v2family default; TP4canary-1b-v2-tp4tests/e2e/models/canary/manifests/canary-1b-v2-tp4.json

Family-owned configuration

This family does not declare a family-owned --set namespace. Use the explicit CLI options shown below and the shared configuration namespaces documented in Configure Runtime Behavior.

Family-specific CLI contracts

Inputs and options below are filtered by the declared E2E task and the methods and configuration fields used by that family's native runtime implementation. The global CLI parser accepts a wider union of flags; flags absent here are not declared for this family.

trtmc transcribe

Transcribe an audio file.

Declared recipes: canary-1b-v2, canary-1b-v2-tp4

trtmc transcribe <bundle.bundle> --audio <input.wav>
Supported input or optionRequirementRuntime behavior
--audio <PATH>RequiredInput WAV file; repeat for supported offline batches.
--max-new-tokens <N>OptionalSet the maximum number of decoder output tokens.
--beam-size <N>OptionalSelect greedy decoding at 1 or beam search above 1.
--length-penalty <F>OptionalSet the beam-search length-normalization exponent.
--beam-fallback-max-size <N>OptionalRetry an unterminated decode with larger beams up to N.
--source-language <TAG>OptionalSet the source language for a multilingual transcription request.
--target-language <TAG>OptionalSet the target language for a multilingual transcription request.
--task <transcribe|translate>OptionalSelect transcription or speech translation.
--punctuation | --no-punctuationOptionalEnable or disable punctuation in the result.
--timestampsOptionalReturn timestamped transcription segments.
--max-input-seconds <F>OptionalReject input longer than the specified duration.
--segment-length-seconds <F>OptionalSplit long audio into segments no longer than this duration.
--segment-min-seconds <F>OptionalSet the minimum duration for dynamic segmentation windows.
--segment-overlap-seconds <F>OptionalSet overlap between adjacent transcription segments.
--lcs-mergeOptionalMerge overlapping segment tokens with boundary-constrained LCS.
Code basis

Runtime provider: canary

  • include/trtmc/pipeline.h
  • src/cli/main.cpp
  • src/runtime/models/canary/canary_request.h
  • src/runtime/models/canary/pipeline.cpp
  • tests/e2e/models/canary/manifests/canary-1b-v2-tp4.json
  • tests/e2e/models/canary/manifests/canary-1b-v2.json

Bundle lifecycle commands

These commands apply to every declared recipe in this family; they do not add task inputs or model capabilities.

trtmc build

Build one exact checkpoint into a TensorRT-Model-Connect bundle.

trtmc build <hf-id> --output <bundle.bundle>

trtmc inspect

Inspect bundle metadata, runtime identity, and packaged sections.

trtmc inspect <bundle.bundle>

See the CLI Reference for all options and limitations.