higher precision for vae

patrickvonplaten · patrickvonplaten · commit 81b6fbf19d11 · 2022-10-28T18:19:06.000Z
diff --git a/tests/models/test_models_vae.py b/tests/models/test_models_vae.py
@@ -209,7 +209,7 @@ def test_stable_diffusion_fp16(self, seed, expected_slice):
         output_slice = sample[-1, -2:, :2, -2:].flatten().float().cpu()
         expected_output_slice = torch.tensor(expected_slice)
 
-        assert torch_all_close(output_slice, expected_output_slice, atol=5e-3)
+        assert torch_all_close(output_slice, expected_output_slice, atol=1e-2)
 
     @parameterized.expand(
         [

Original file line number	Diff line number	Diff line change
`@@ -209,7 +209,7 @@ def test_stable_diffusion_fp16(self, seed, expected_slice):`
`209`	`209`	`output_slice = sample[-1, -2:, :2, -2:].flatten().float().cpu()`
`210`	`210`	`expected_output_slice = torch.tensor(expected_slice)`
`211`	`211`
`212`		`- assert torch_all_close(output_slice, expected_output_slice, atol=5e-3)`
	`212`	`+ assert torch_all_close(output_slice, expected_output_slice, atol=1e-2)`
`213`	`213`
`214`	`214`	`@parameterized.expand(`
`215`	`215`	`[`