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