Mirrors
/
diffusers
mirror of https://github.com/ShivamShrirao/diffusers.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

higher precision for vae

This commit is contained in:
Patrick von Platen 2022-10-28 18:19:06 +00:00
parent a7ae808ee2
commit 81b6fbf19d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/test_models_vae.py
View File

@ -209,7 +209,7 @@ class AutoencoderKLIntegrationTests(unittest.TestCase):
output_slice = sample[-1, -2:, :2, -2:].flatten().float().cpu() output_slice = sample[-1, -2:, :2, -2:].flatten().float().cpu()
expected_output_slice = torch.tensor(expected_slice) 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( @parameterized.expand(
[ [