actually fix test_ldm_text2img_fast

9b9afc97 · patil-suraj · b7f0ce5b · 9b9afc97
Commit 9b9afc97 authored Jun 27, 2022 by patil-suraj
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/test_modeling_utils.py tests/test_modeling_utils.py +1 -1

No files found.
--- a/tests/test_modeling_utils.py
+++ b/tests/test_modeling_utils.py
@@ -701,7 +701,7 @@ class PipelineTesterMixin(unittest.TestCase):

        prompt = "A painting of a squirrel eating a burger"
        generator = torch.manual_seed(0)
-        image = ldm([prompt], generator=generator, num_inference_steps=20)
+        image = ldm([prompt], generator=generator, num_inference_steps=1)

        image_slice = image[0, -1, -3:, -3:].cpu()