Added safetensor dependency for nvidia bert

ctuning · Aug 8, 2023 · b0c62fc · b0c62fc
1 parent 352bc2b
commit b0c62fc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/mlperf/inference/bert/run_sparse_models.sh b/docs/mlperf/inference/bert/run_sparse_models.sh
@@ -42,7 +42,7 @@ cmd="cm run script --tags=run,mlperf,inference,generate-run-cmds,_find-performan
    --mode=performance \
    --execution_mode=valid \
    --adr.mlperf-inference-implementation.max_batchsize=384 \
-   --power=yes --adr.mlperf-power-client.power_server=192.168.0.15
+   --power=yes --adr.mlperf-power-client.power_server=192.168.0.15 --adr.mlperf-power-client.port=4940
    --results_dir=$HOME/results_dir \
    --env.CM_MLPERF_NEURALMAGIC_MODEL_ZOO_STUB=$stub \
    --quiet"