mirror of
https://github.com/huggingface/diffusers.git
synced 2026-01-27 17:22:53 +03:00
fix tests
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user