Skip to content

GitLab

Commit 69c76173 authored by Patrick von Platen's avatar Patrick von Platen
Browse files

fix tests

parent 926b34b4
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/pipelines/dit/test_dit.py
View file @ 69c76173
...@@ -111,7 +111,7 @@ class DiTPipelineIntegrationTests(unittest.TestCase): ...@@ -111,7 +111,7 @@ class DiTPipelineIntegrationTests(unittest.TestCase):
expected_image = load_numpy( expected_image = load_numpy(
f"https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/dit/{word}.npy" 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): def test_dit_512_fp16(self):
generator = torch.manual_seed(0) generator = torch.manual_seed(0)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment