1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00

fix tests

This commit is contained in:
Patrick von Platen
2023-01-22 14:31:05 +02:00
parent 926b34b40c
commit 69c76173fa

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).abs()) < 1e-3
assert np.abs((expected_image - image).max()) < 1e-3
def test_dit_512_fp16(self):
generator = torch.manual_seed(0)