diff --git a/.github/workflows/test_models.yml b/.github/workflows/test_models.yml index 88218690e..dc8176b55 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 pytest -v models/turbine_models/tests/sdxl_test.py --device rocm --rt_device hip --iree_target_triple gfx90a --precision fp16 --attn_spec default - pytest -v models/turbine_models/tests/sdxl_test.py --device rocm --rt_device hip --iree_target_triple gfx90a --precision fp16 --attn_spec default --batch_size 2 + pytest -v models/turbine_models/tests/sdxl_test.py --device rocm --rt_device hip --iree_target_triple gfx90a --precision fp16 --clip_spec None --unet_spec default --vae_spec default --batch_size 2 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