Skip to content

Commit 9808a1b

Browse files
authored
Update target directory path for NVIDIA code
1 parent ea96776 commit 9808a1b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

script/add-custom-nvidia-system/customize.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ def postprocess(i):
3434
mlc_model = env["MLC_MODEL"]
3535
if "llama2-70b" in mlc_model:
3636
mlc_model = "llama2-70b"
37-
target_dir = os.path.join(mlc_code_path, "configs", system_name, scenario)
37+
target_dir = os.path.join(env['MLC_MLPERF_INFERENCE_NVIDIA_CODE_PATH'], "configs", system_name, scenario)
3838
os.makedirs(target_dir, exist_ok=True)
3939
src_file = os.path.join(tmp_script_path, "dummy_config.py")
4040

0 commit comments

Comments
 (0)