Skip to content

Commit

Permalink
Fix failing VAE tiling test (#7747)
Browse files Browse the repository at this point in the history
update
  • Loading branch information
DN6 authored Apr 24, 2024
1 parent 7404f1e commit 88018fc
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/pipelines/test_pipelines_common.py
Original file line number Diff line number Diff line change
Expand Up @@ -120,7 +120,7 @@ def test_vae_tiling(self):
inputs["return_dict"] = False
output_2 = pipe(**inputs)[0]

assert np.abs(output_2 - output_1).max() < 5e-1
assert np.abs(to_np(output_2) - to_np(output_1)).max() < 5e-1

# test that tiled decode works with various shapes
shapes = [(1, 4, 73, 97), (1, 4, 97, 73), (1, 4, 49, 65), (1, 4, 65, 49)]
Expand Down

0 comments on commit 88018fc

Please sign in to comment.