Skip to content

Commit cfda7cd

Browse files
committed
[Automated Commit] Format Codebase
1 parent c32d3a5 commit cfda7cd

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

script/app-mlperf-inference-mlcommons-python/customize.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -499,7 +499,7 @@ def get_run_cmd_reference(
499499

500500
if env.get('CM_ACTIVATE_RGAT_IN_MEMORY', '') == "yes":
501501
cmd += " --in-memory "
502-
502+
503503
elif "llama3" in env['CM_MODEL']:
504504
env['RUN_DIR'] = os.path.join(
505505
env['CM_MLPERF_INFERENCE_SOURCE'],
@@ -520,12 +520,10 @@ def get_run_cmd_reference(
520520

521521
if env.get('CM_MLPERF_INFERENCE_NUM_WORKERS', '') != '':
522522
cmd += f" --num-workers {env['CM_MLPERF_INFERENCE_NUM_WORKERS']}"
523-
524-
523+
525524
cmd = cmd.replace("--count", "--total-sample-count")
526525
cmd = cmd.replace("--max-batchsize", "--batch-size")
527526

528-
529527
if env.get('CM_NETWORK_LOADGEN', '') in ["lon", "sut"]:
530528
cmd = cmd + " " + "--network " + env['CM_NETWORK_LOADGEN']
531529
if env.get('CM_NETWORK_LOADGEN_SUT_SERVERS', []):

0 commit comments

Comments
 (0)