update

2025-12-08 13:34:27 +08:00 · 2024-04-22 09:31:31 +00:00
1 changed files with 1 additions and 1 deletions
--- a/tests/pipelines/test_pipelines_common.py
+++ b/tests/pipelines/test_pipelines_common.py
@@ -120,7 +120,7 @@ class SDFunctionTesterMixin:
        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)]