TST Fewer inference steps for stable diffusion (huggingface#2051)

Reduce the number of inference steps for stable diffusion tests. These tests are the slowest ones on CI, this should help (~3 min on average).
dengdifan · Sep 6, 2024 · b180ae4 · b180ae4
1 parent 31fbbd2
commit b180ae4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_stablediffusion.py b/tests/test_stablediffusion.py
@@ -145,7 +145,7 @@ def instantiate_sd_peft(self, model_id, config_cls, config_kwargs):
     def prepare_inputs_for_testing(self):
         return {
             "prompt": "a high quality digital photo of a cute corgi",
-            "num_inference_steps": 20,
+            "num_inference_steps": 3,
         }
 
     @parameterized.expand(