From a6171cbe71eaacfb249b51f8d27b6065b82c5765 Mon Sep 17 00:00:00 2001 From: chenyu Date: Tue, 26 Nov 2024 22:09:43 -0500 Subject: [PATCH] add stable diffusion v2 to mac benchmark (#7917) this caught #7902 --- .github/workflows/benchmark.yml | 3 +++ 1 file changed, 3 insertions(+) diff --git a/.github/workflows/benchmark.yml b/.github/workflows/benchmark.yml index 43d81f23ec..94285dd1af 100644 --- a/.github/workflows/benchmark.yml +++ b/.github/workflows/benchmark.yml @@ -52,6 +52,8 @@ jobs: run: JIT=1 python3.11 examples/stable_diffusion.py --fp16 --seed 0 --noshow --timing | tee sd.txt - name: Run Stable Diffusion without fp16 run: JIT=1 python3.11 examples/stable_diffusion.py --seed 0 --noshow --timing | tee sd_no_fp16.txt + - name: Run Stable Diffusion v2 + run: JIT=1 python3.11 examples/sdv2.py --fp16 --seed 0 --noshow --timing | tee sdv2.txt - name: Run SDXL run: JIT=1 python3.11 examples/sdxl.py --seed 0 --noshow --timing | tee sdxl.txt - name: Run model inference benchmark @@ -119,6 +121,7 @@ jobs: matmul_bfloat16.txt sd.txt sd_no_fp16.txt + sdv2.txt sdxl.txt beautiful_mnist.txt train_cifar.txt