Update run.sh

mlcommons · Oct 21, 2024 · 8e9bf95 · 8e9bf95
1 parent f0d8e1f
commit 8e9bf95
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/run.sh b/run.sh
@@ -1 +1,14 @@
-INFERENCE_RESULTS_VERSION=v4.1 INFERENCE_RESULTS_REPO_OWNER=GATEOverflow INFERENCE_RESULTS_REPO_BRANCH=main INFERENCE_RESULTS_REPO_NAME=mlperf_inference_submissions bash docinit.sh
+INFERENCE_RESULTS_REPO_OWNER=${INFERENCE_RESULTS_REPO_OWNER:-mlcommons}
+INFERENCE_RESULTS_REPO_NAME=${INFERENCE_RESULTS_REPO_NAME:-mlperf_inference_submissions}
+INFERENCE_RESULTS_REPO_BRANCH=${INFERENCE_RESULTS_REPO_BRANCH:-main}
+INFERENCE_RESULTS_VERSION=${INFERENCE_RESULTS_VERSION:-v4.1}
+
+if [ ! -e docs ]; then
+    git clone https://github.com/GATEOverflow/inference_results_visualization_template.git docs
+    test $? -eq 0 || exit $?
+fi
+
+cp docs/docinit.sh .
+
+
+INFERENCE_RESULTS_VERSION=${INFERENCE_RESULTS_VERSION} INFERENCE_RESULTS_REPO_OWNER=${INFERENCE_RESULTS_REPO_OWNER} INFERENCE_RESULTS_REPO_BRANCH=${INFERENCE_RESULTS_REPO_BRANCH} INFERENCE_RESULTS_REPO_NAME=${INFERENCE_RESULTS_REPO_NAME} bash docinit.sh