Mirrors
/
diffusers
mirror of https://github.com/ShivamShrirao/diffusers.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

improve tests

This commit is contained in:
Patrick von Platen 2023-01-22 14:30:15 +02:00
parent 8d326e61cf
commit 926b34b40c
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/pipelines/dit/test_dit.py
View File

@ -111,7 +111,7 @@ class DiTPipelineIntegrationTests(unittest.TestCase):
expected_image = load_numpy(
f"https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/dit/{word}.npy"
)
assert np.abs((expected_image - image).sum()) < 1e-3
assert np.abs((expected_image - image).abs()) < 1e-3
def test_dit_512_fp16(self):
generator = torch.manual_seed(0)
@ -130,4 +130,4 @@ class DiTPipelineIntegrationTests(unittest.TestCase):
"https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main"
f"/dit/{word}_fp16.npy"
)
assert np.abs((expected_image - image).sum()) < 1e-3
assert np.abs((expected_image - image).max()) < 1e-2

4
tests/pipelines/stable_diffusion/test_stable_diffusion_instruction_pix2pix.py
View File

@ -307,13 +307,13 @@ class StableDiffusionInstructPix2PixPipelineSlowTests(unittest.TestCase):
assert latents.shape == (1, 4, 64, 64)
latents_slice = latents[0, -3:, -3:, -1]
expected_slice = np.array([-0.2388, -0.4673, -0.9775, 1.5127, 1.4414, 0.7778, 0.9907, 0.8472, 0.7788])
assert np.abs(latents_slice.flatten() - expected_slice).max() < 1e-3
assert np.abs(latents_slice.flatten() - expected_slice).max() < 1e-2
elif step == 2:
latents = latents.detach().cpu().numpy()
assert latents.shape == (1, 4, 64, 64)
latents_slice = latents[0, -3:, -3:, -1]
expected_slice = np.array([-0.2568, -0.4648, -0.9639, 1.5137, 1.4609, 0.7603, 0.9795, 0.8403, 0.7949])
assert np.abs(latents_slice.flatten() - expected_slice).max() < 1e-3
assert np.abs(latents_slice.flatten() - expected_slice).max() < 1e-2
callback_fn.has_been_called = False