From 9b9afc9726ef1656552bc7bfa2e5afac696b2070 Mon Sep 17 00:00:00 2001 From: patil-suraj Date: Mon, 27 Jun 2022 11:46:50 +0200 Subject: [PATCH] actually fix test_ldm_text2img_fast --- tests/test_modeling_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_modeling_utils.py b/tests/test_modeling_utils.py index ea94f48608..697a377f8c 100755 --- 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()