Github action diffusers test case change

huggingface · Oct 19, 2024 · 3104e01 · 3104e01
1 parent f98688d
commit 3104e01
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/ci/slow_tests_diffusers.sh b/tests/ci/slow_tests_diffusers.sh
@@ -1,7 +1,10 @@
 #!/bin/bash
 
+hl-smi
+hl-smi -q
+
 python -m pip install --upgrade pip
 export RUN_SLOW=true
 make test_installs
-CUSTOM_BF16_OPS=1 python -m pytest tests/test_diffusers.py -v -s -k "test_no_throughput_regression_autocast"
+CUSTOM_BF16_OPS=1 python -m pytest tests/test_diffusers.py -v -s -k "test_stable_diffusion_default"
 make slow_tests_diffusers