We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent f28aee3 commit b88bbcdCopy full SHA for b88bbcd
tools/perf/benchmark.sh
@@ -92,7 +92,7 @@ echo "Benchmarking SD UNet model"
92
for bs in ${large_model_batch_sizes[@]}
93
do
94
python perf_run.py --model_torch sd_unet \
95
- --precision fp32,fp16 --inputs="(${bs}, 4, 128, 128)@fp16;(${bs})@fp16;(${bs}, 1, 768)@fp16" \
+ --precision fp32,fp16 --inputs="(${bs}, 4, 64, 64)@fp16;(${bs})@fp16;(${bs}, 1, 768)@fp16" \
96
--batch_size ${bs} \
97
--backends torch,dynamo,torch_compile,inductor \
98
--truncate \
0 commit comments