Commit 41db2dbf authored by Patrick von Platen's avatar Patrick von Platen
Browse files

correct tests

parent a7ca03aa
...@@ -188,7 +188,7 @@ class StableDiffusionLatentUpscalePipelineIntegrationTests(unittest.TestCase): ...@@ -188,7 +188,7 @@ class StableDiffusionLatentUpscalePipelineIntegrationTests(unittest.TestCase):
expected_image = load_numpy( expected_image = load_numpy(
"https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/latent-upscaler/astronaut_1024.npy" "https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/latent-upscaler/astronaut_1024.npy"
) )
assert np.abs((expected_image - image).max()) < 5e-2 assert np.abs((expected_image - image).max()) < 5e-1
def test_latent_upscaler_fp16_image(self): def test_latent_upscaler_fp16_image(self):
generator = torch.manual_seed(33) generator = torch.manual_seed(33)
...@@ -216,4 +216,4 @@ class StableDiffusionLatentUpscalePipelineIntegrationTests(unittest.TestCase): ...@@ -216,4 +216,4 @@ class StableDiffusionLatentUpscalePipelineIntegrationTests(unittest.TestCase):
expected_image = load_numpy( expected_image = load_numpy(
"https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/latent-upscaler/fire_temple_1024.npy" "https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/latent-upscaler/fire_temple_1024.npy"
) )
assert np.abs((expected_image - image).max()) < 5e-1 assert np.abs((expected_image - image).max()) < 5e-2
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