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

correct tests

Browse Source
This commit is contained in:
Patrick von Platen
2023-02-08 11:12:51 +02:00
parent a7ca03aa85
commit 41db2dbf90
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/pipelines/stable_diffusion_2/test_stable_diffusion_latent_upscale.py
View File

@@ -188,7 +188,7 @@ class StableDiffusionLatentUpscalePipelineIntegrationTests(unittest.TestCase):
expected_image = load_numpy(
"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):
generator = torch.manual_seed(33)
@@ -216,4 +216,4 @@ class StableDiffusionLatentUpscalePipelineIntegrationTests(unittest.TestCase):
expected_image = load_numpy(
"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