Skip to content

Commit e57b755

Browse files
committed
add arg to override default repo tag for all dependencies
1 parent 378d59a commit e57b755

1 file changed

Lines changed: 8 additions & 0 deletions

File tree

build.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2738,6 +2738,12 @@ def enable_all():
27382738
default=DEFAULT_TRITON_VERSION_MAP["upstream_container_version"],
27392739
help="This flag sets the upstream container version for Triton Inference Server to be built. Default: the latest released version.",
27402740
)
2741+
parser.add_argument(
2742+
"--default-repo-tag",
2743+
required=False,
2744+
default=None,
2745+
help="Override the calculated default-repo-tag value",
2746+
)
27412747
parser.add_argument(
27422748
"--ort-version",
27432749
required=False,
@@ -2866,6 +2872,8 @@ def enable_all():
28662872
cver = FLAGS.triton_container_version
28672873
if not cver.endswith("dev"):
28682874
default_repo_tag = "r" + cver
2875+
if FLAGS.default_repo_tag:
2876+
default_repo_tag = FLAGS.default_repo_tag
28692877
log("default repo-tag: {}".format(default_repo_tag))
28702878

28712879
# For other versions use the TRITON_VERSION_MAP unless explicitly

0 commit comments

Comments
 (0)