test changes reverted

anandhu-eng · Jul 8, 2024 · 0bd277a · 0bd277a
1 parent 0247b0a
commit 0bd277a
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 7 deletions.
diff --git a/script/app-mlperf-inference-nvidia/_cm.yaml b/script/app-mlperf-inference-nvidia/_cm.yaml
@@ -560,7 +560,6 @@ variations:
 
   llama2-70b_:
     deps:
-      #- tags: get,ml-model,llama2-70b,_nvidia,_fp8
       - tags: get,generic-python-lib,_package.transformers
         names:
           - transformers

diff --git a/script/get-ml-model-llama2/_cm.json b/script/get-ml-model-llama2/_cm.json
@@ -51,6 +51,7 @@
   "variations": {
     "meta-llama/Llama-2-70b-chat-hf": {
       "group": "huggingface-stub",
+      "default": true,
       "env": {
         "CM_GIT_CHECKOUT_FOLDER": "Llama-2-70b-chat-hf",
         "CM_MODEL_ZOO_ENV_KEY": "LLAMA2"
@@ -63,7 +64,6 @@
     },
     "meta-llama/Llama-2-7b-chat-hf": {
       "group": "huggingface-stub",
-      "default": true,
       "env": {
         "CM_GIT_CHECKOUT_FOLDER": "Llama-2-7b-chat-hf",
         "CM_MODEL_ZOO_ENV_KEY": "LLAMA2"

diff --git a/script/get-ml-model-llama2/run-nvidia.sh b/script/get-ml-model-llama2/run-nvidia.sh
@@ -1,6 +1,5 @@
 #!/bin/bash
 
-<<COMMENT
 if [[ ! -e ${CM_NVIDIA_MLPERF_SCRATCH_PATH}/models/Llama2/Llama-2-70b-chat-hf ]]; then
   mkdir -p ${CM_NVIDIA_MLPERF_SCRATCH_PATH}/models/Llama2/Llama-2-70b-chat-hf
   cd ${LLAMA2_CHECKPOINT_PATH}
@@ -20,7 +19,3 @@ export DOCKER_RUN_ARGS="$DOCKER_RUN_ARGS"
 export RUN_CMD="$RUN_CMD"
 make -C docker run LOCAL_USER=1
 test $? -eq 0 || exit $?
-COMMENT
-
-echo ${CM_NVIDIA_MLPERF_SCRATCH_PATH}
-