Skip to content

Commit 41db2db

Browse files
correct tests
1 parent a7ca03a commit 41db2db

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/pipelines/stable_diffusion_2/test_stable_diffusion_latent_upscale.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -188,7 +188,7 @@ def test_latent_upscaler_fp16(self):
188188
expected_image = load_numpy(
189189
"https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/latent-upscaler/astronaut_1024.npy"
190190
)
191-
assert np.abs((expected_image - image).max()) < 5e-2
191+
assert np.abs((expected_image - image).max()) < 5e-1
192192

193193
def test_latent_upscaler_fp16_image(self):
194194
generator = torch.manual_seed(33)
@@ -216,4 +216,4 @@ def test_latent_upscaler_fp16_image(self):
216216
expected_image = load_numpy(
217217
"https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/latent-upscaler/fire_temple_1024.npy"
218218
)
219-
assert np.abs((expected_image - image).max()) < 5e-1
219+
assert np.abs((expected_image - image).max()) < 5e-2

0 commit comments

Comments
 (0)