Skip to content

Commit b0ff02d

Browse files
committed
add arg to override default repo tag for all dependencies
1 parent 23869bb commit b0ff02d

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
@@ -2739,6 +2739,12 @@ def enable_all():
27392739
default=DEFAULT_TRITON_VERSION_MAP["upstream_container_version"],
27402740
help="This flag sets the upstream container version for Triton Inference Server to be built. Default: the latest released version.",
27412741
)
2742+
parser.add_argument(
2743+
"--default-repo-tag",
2744+
required=False,
2745+
default=None,
2746+
help="Override the calculated default-repo-tag value",
2747+
)
27422748
parser.add_argument(
27432749
"--ort-version",
27442750
required=False,
@@ -2867,6 +2873,8 @@ def enable_all():
28672873
cver = FLAGS.triton_container_version
28682874
if not cver.endswith("dev"):
28692875
default_repo_tag = "r" + cver
2876+
if FLAGS.default_repo_tag:
2877+
default_repo_tag = FLAGS.default_repo_tag
28702878
log("default repo-tag: {}".format(default_repo_tag))
28712879

28722880
# For other versions use the TRITON_VERSION_MAP unless explicitly

0 commit comments

Comments
 (0)