Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 6 additions & 0 deletions python/tensorrt_model_connect/families/canary/MODEL.toml
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,12 @@

id = "canary"
plugin = "canary"
python_profile_specs = [
"canary_reference|families/canary/python_profile_requirements/canary_reference.lock.txt|families/canary/python_profile_verify.py|true|true",
]
default_execution_profiles = [
"reference|canary_reference",
]
module = "plugin"
model_dir_adapter = "nemo_archive.py|resolve_model_dir"
nemo_archive_adapter = "nemo_archive.py|resolve_nemo_archive"
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,109 @@
absl-py==2.5.0
annotated-types==0.7.0
antlr4-python3-runtime==4.9.3
asttokens==3.0.2
audioread==3.1.0
braceexpand==0.1.7
cffi==2.1.1
cloudpickle==3.1.2
contourpy==1.3.3
cycler==0.12.1
cytoolz==1.1.0
datasets==3.6.0
decorator==5.3.1
dill==0.3.8
docopt==0.6.2
editdistance==0.8.1
einops==0.8.2
executing==2.2.1
fiddle==0.3.0
fonttools==4.63.0
fsspec==2024.12.0
graphviz==0.21
grpcio==1.76.0
huggingface-hub==0.36.2
hydra-core==1.3.2
intervaltree==3.2.1
ipython-pygments-lexers==1.1.1
ipython==9.16.1
jedi==0.20.0
jiwer==3.1.0
joblib==1.5.3
kaldi-python-io==1.2.2
kaldialign==0.9.1
kiwisolver==1.5.0
lazy-loader==0.5
lhotse==1.33.0
libcst==1.9.0
librosa==0.11.0
lightning-utilities==0.15.3
lightning==2.4.0
llvmlite==0.43.0
markdown==3.10.3
marshmallow==4.3.1
matplotlib-inline==0.2.2
matplotlib==3.11.1
msgpack==1.2.1
multiprocess==0.70.16
nemo_toolkit[asr]==2.7.3
numba==0.60.0
numexpr==2.10.0
numpy==1.26.4
nv-one-logger-core==2.3.1
nv-one-logger-pytorch-lightning-integration==2.3.1
nv-one-logger-training-telemetry==2.3.1
omegaconf==2.3.0
overrides==7.7.0
pandas==2.2.3
parso==0.8.7
platformdirs==4.11.2
pooch==1.9.0
prompt-toolkit==3.0.53
protobuf==5.29.5
pure-eval==0.2.3
pyannote.core==5.0.0
pyannote.database==5.1.3
pyannote.metrics==3.2.1
pyarrow==20.0.0
pycparser==3.0
pydantic-core==2.27.2
pydantic==2.10.6
pydub==0.25.1
pyloudnorm==0.2.0
pyparsing==3.3.2
python-dateutil==2.9.0.post0
pytorch-lightning==2.4.0
pytz==2026.3.post1
rapidfuzz==3.14.5
resampy==0.4.3
ruamel.yaml.clib==0.2.12
ruamel.yaml==0.18.10
scikit-learn==1.4.2
scipy==1.15.3
sentencepiece==0.2.2
silero-vad==6.2.1
six==1.17.0
sortedcontainers==2.4.0
soundfile==0.14.0
sox==1.5.0
soxr==0.5.0.post1
stack-data==0.6.3
strenum==0.4.15
tensorboard-data-server==0.7.2
tensorboard==2.20.0
text-unidecode==1.3
threadpoolctl==3.6.0
tokenizers==0.22.2
toml==0.10.2
toolz==1.1.0
torchmetrics==1.9.0
traitlets==5.16.1
transformers==4.57.6
typing-extensions==4.15.0
tzdata==2026.3
webdataset==1.0.2
werkzeug==3.1.8
wget==3.2
whisper-normalizer==0.1.15
wrapt==2.3.0
xxhash==3.8.1
Original file line number Diff line number Diff line change
@@ -0,0 +1,43 @@
# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0

from importlib.metadata import version

import absl
import cffi
import datasets
import librosa
import nemo.collections.asr as nemo_asr
import numpy
import pandas
import pycparser
import pydantic
import sentencepiece
import soundfile
import transformers
from nemo.collections.asr.models import ASRModel

assert version("absl-py") == "2.5.0"
assert version("cffi") == "2.1.1"
assert version("datasets") == "3.6.0"
assert version("librosa") == "0.11.0"
assert version("nemo_toolkit") == "2.7.3"
assert version("numpy") == "1.26.4"
assert version("pandas") == "2.2.3"
assert version("soundfile") == "0.14.0"
assert version("pycparser") == "3.0"
assert version("pydantic") == "2.10.6"
assert version("sentencepiece") == "0.2.2"
assert transformers.__version__ == "4.57.6"
assert nemo_asr is not None
assert ASRModel is not None
assert numpy.__version__ == "1.26.4"
assert pandas.__version__ == "2.2.3"
assert soundfile.__version__ == "0.14.0"
assert absl is not None
assert cffi is not None
assert datasets is not None
assert librosa is not None
assert pycparser is not None
assert pydantic is not None
assert sentencepiece is not None
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,13 @@
id = "deepseek_v2"
plugin = "deepseek_v2"
module = "plugin"
python_profile_specs = [
"deepseek_v2|families/deepseek_v2/python_profile_requirements/deepseek_v2.lock.txt|families/deepseek_v2/python_profile_verify.py|true",
]
default_execution_profiles = [
"build|deepseek_v2",
"reference|deepseek_v2",
]
capabilities = ["debug_layer_outputs"]
aliases = [
"deepseek_v2",
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,6 @@
# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0

transformers==5.2.0
tokenizers==0.22.2
huggingface-hub==1.22.0
Original file line number Diff line number Diff line change
@@ -0,0 +1,30 @@
# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0

"""Verify the DeepSeek-V2 tokenizer-compatible Python environment."""

from importlib.metadata import version

import transformers
from transformers import (
DeepseekV2Config,
DeepseekV2ForCausalLM,
DeepseekV3Config,
DeepseekV3ForCausalLM,
)


assert transformers.__version__ == "5.2.0"
assert version("tokenizers") == "0.22.2"
assert version("huggingface-hub") == "1.22.0"
assert DeepseekV2Config.model_type == "deepseek_v2"
assert DeepseekV2ForCausalLM.config_class is DeepseekV2Config
assert DeepseekV3ForCausalLM.config_class is DeepseekV3Config
deepseek_v3_config = DeepseekV3Config(experts_implementation="batched_mm")
assert deepseek_v3_config._experts_implementation == "batched_mm"
print(
f"transformers={transformers.__version__} "
f"tokenizers={version('tokenizers')} "
f"huggingface-hub={version('huggingface-hub')} "
"deepseek_v2=available experts_implementation=batched_mm"
)
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,12 @@
id = "minimax_h3"
plugin = "minimax_h3"
module = "plugin"
python_profile_specs = [
"minimax_h3_reference|families/minimax_h3/python_profile_requirements/minimax_h3_reference.lock.txt|families/minimax_h3/python_profile_reference_verify.py|true|true",
]
default_execution_profiles = [
"reference|minimax_h3_reference",
]
aliases = [
"minimax-h3",
"minimax_h3",
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0

"""Verify the pinned MiniMax-H3 reference execution profile."""

from importlib.metadata import version

from huggingface_hub import get_cached_repo_tree


assert version("huggingface-hub") == "1.23.0"
assert callable(get_cached_repo_tree)
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0

huggingface-hub==1.23.0
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,12 @@

id = "nemotron_labs_diffusion"
plugin = "nemotron_labs_diffusion"
python_profile_specs = [
"nemotron_labs_diffusion_reference|families/nemotron_labs_diffusion/python_profile_requirements/reference.lock.txt|families/nemotron_labs_diffusion/python_profile_reference_verify.py|true",
]
default_execution_profiles = [
"reference|nemotron_labs_diffusion_reference",
]
module = "plugin"
aliases = [
"nemotron_labs_diffusion",
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,24 @@
# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0

"""Verify the Nemotron Labs Diffusion LoRA reference profile."""

from importlib.metadata import version

import accelerate
import peft
import safetensors
import torch
import transformers


assert version("accelerate") == "1.14.0"
assert version("huggingface-hub") == "1.22.0"
assert version("peft") == "0.20.0"
assert version("safetensors") == "0.8.0"
assert version("tokenizers") == "0.22.2"
assert transformers.__version__ == "5.2.0"
assert accelerate is not None
assert peft is not None
assert safetensors is not None
assert callable(torch.cuda.is_available)
Original file line number Diff line number Diff line change
@@ -0,0 +1,9 @@
# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0

accelerate==1.14.0
huggingface-hub==1.22.0
peft==0.20.0
safetensors==0.8.0
tokenizers==0.22.2
transformers==5.2.0
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,12 @@

id = "nemotron_speech_streaming"
plugin = "nemotron_speech_streaming"
python_profile_specs = [
"nemotron_speech_streaming_reference|families/nemotron_speech_streaming/python_profile_requirements/nemotron_speech_streaming_reference.lock.txt|families/nemotron_speech_streaming/python_profile_verify.py|true|true",
]
default_execution_profiles = [
"reference|nemotron_speech_streaming_reference",
]
module = "plugin"
model_dir_adapter = "nemo_archive.py|resolve_model_dir"
nemo_archive_adapter = "nemo_archive.py|resolve_nemo_archive"
Expand Down
Loading
Loading