Update run_benchmark.sh

Signed-off-by: Wang, Mengni <[email protected]>
onnx · Aug 16, 2024 · 77db4f3 · 77db4f3
1 parent 206bf96
commit 77db4f3
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/...ngface_model/text_to_image/stable_diffusion_v1_5/quantization/ptq_static/run_benchmark.sh b/...ngface_model/text_to_image/stable_diffusion_v1_5/quantization/ptq_static/run_benchmark.sh
@@ -56,12 +56,11 @@ function run_benchmark {
 	extra_cmd=$extra_cmd"--image_path=${image_path} "
     fi
 
-    python main.py \
-            --model_path="${input_model}" \
-            --batch_size="${batch_size-1}" \
-            --benchmark \
-	    ${extra_cmd}
-
+    if [ "$batch_size" ]; then
+	extra_cmd=$extra_cmd"--batch_size=${batch_size} "
+    fi
+    extra_cmd=$extra_cmd"--benchmark"
+    eval "python main.py --model_path=${input_model} ${extra_cmd}"
 }
 
 main "$@"