Fix for nvidia gptj

anandhu-eng · May 24, 2024 · ef2a9fe · ef2a9fe
1 parent b35dfaa
commit ef2a9fe
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/script/app-mlperf-inference-nvidia/customize.py b/script/app-mlperf-inference-nvidia/customize.py
@@ -166,7 +166,8 @@ def preprocess(i):
 
         if not os.path.exists(fp32_model_path):
             env['CM_REQUIRE_GPTJ_MODEL_DOWNLOAD'] = 'yes' # download via prehook_deps
-            cmds.append(f"cp -r $CM_ML_MODEL_FILE_WITH_PATH {fp32_model_path}")
+            if make_command == "build_engine":
+                cmds.append(f"cp -r $CM_ML_MODEL_FILE_WITH_PATH {fp32_model_path}")
 
         model_name = "gptj"
         model_path = fp8_model_path