Skip to content

Commit

Permalink
added docker_cm_repo_branch to more run option - docker
Browse files Browse the repository at this point in the history
  • Loading branch information
anandhu-eng committed Oct 6, 2024
1 parent af2780c commit ed593ba
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions main.py
Original file line number Diff line number Diff line change
Expand Up @@ -179,6 +179,7 @@ def mlperf_inference_implementation_readme(spaces, model, implementation, *, imp
content += f"{cur_space3}<details>\n"
content += f"{cur_space3}<summary> Please click here to see more options for the docker launch </summary>\n\n"
content += f"{cur_space3}* `--docker_cm_repo=<Custom CM GitHub repo URL in username@repo format>`: to use a custom fork of cm4mlops repository inside the docker image\n\n"
content += f"{cur_space3}* `--docker_cm_repo_branch=<Custom CM GitHub repo Branch>`: to checkout a custom branch of the cloned cm4mlops repository inside the docker image\n\n"
content += f"{cur_space3}* `--docker_cache=no`: to not use docker cache during the image build\n"

if implementation.lower() == "nvidia":
Expand Down

0 comments on commit ed593ba

Please sign in to comment.