Skip to content

Commit b88bbcd

Browse files
authored
minor fix: Update Benchmark values (#2453) (#2454)
1 parent f28aee3 commit b88bbcd

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tools/perf/benchmark.sh

+1-1
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,7 @@ echo "Benchmarking SD UNet model"
9292
for bs in ${large_model_batch_sizes[@]}
9393
do
9494
python perf_run.py --model_torch sd_unet \
95-
--precision fp32,fp16 --inputs="(${bs}, 4, 128, 128)@fp16;(${bs})@fp16;(${bs}, 1, 768)@fp16" \
95+
--precision fp32,fp16 --inputs="(${bs}, 4, 64, 64)@fp16;(${bs})@fp16;(${bs}, 1, 768)@fp16" \
9696
--batch_size ${bs} \
9797
--backends torch,dynamo,torch_compile,inductor \
9898
--truncate \

0 commit comments

Comments
 (0)