From fad7e6ef23231795520d8674abc066b826b69674 Mon Sep 17 00:00:00 2001 From: Ean Garvey <87458719+monorimet@users.noreply.github.com> Date: Fri, 13 Sep 2024 20:37:13 -0500 Subject: [PATCH] Update test_models.yml --- .github/workflows/test_models.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/test_models.yml b/.github/workflows/test_models.yml index c7e7c210..550f0f8b 100644 --- a/.github/workflows/test_models.yml +++ b/.github/workflows/test_models.yml @@ -79,6 +79,6 @@ jobs: pytest -v models/turbine_models/tests/sd_test.py pytest -v models/turbine_models/tests/sdxl_test.py --device cpu --rt_device local-task --iree_target_triple x86_64-linux-gnu --num_inference_steps 2 -x pytest -v models/turbine_models/tests/sdxl_test.py --device rocm --rt_device hip --iree_target_triple gfx90a --precision fp16 --attn_spec default -x - pytest -v models/turbine_models/tests/sdxl_test.py --device rocm --rt_device hip --iree_target_triple gfx90a --precision fp16 --attn_spec None --batch_size 2 -x + pytest -v models/turbine_models/tests/sdxl_test.py --device rocm --rt_device hip --iree_target_triple gfx90a --precision fp16 --batch_size 2 -x pytest -v models/turbine_models/tests/sd3_test.py --device cpu --rt_device local-task --iree_target_triple x86_64-linux-gnu --num_inference_steps 2 -x