slice

hlky · hlky · commit e8cd4de028b7 · 2024-12-10T20:52:01.000Z
diff --git a/tests/pipelines/pag/test_pag_sd_inpaint.py b/tests/pipelines/pag/test_pag_sd_inpaint.py
@@ -311,7 +311,7 @@ def test_pag_uncond(self):
         image_slice = image[0, -3:, -3:, -1].flatten()
         assert image.shape == (1, 512, 512, 3)
         expected_slice = np.array(
-            [0.41597816, 0.39302617, 0.44287828, 0.2687074, 0.28315824, 0.40582314, 0.20877528, 0.2380802, 0.39447647]
+            [0.3876953, 0.40356445, 0.4934082, 0.39697266, 0.41674805, 0.41015625, 0.375, 0.36914062, 0.40649414]
         )
         assert (
             np.abs(image_slice.flatten() - expected_slice).max() < 1e-3

Original file line number	Diff line number	Diff line change
`@@ -311,7 +311,7 @@ def test_pag_uncond(self):`
`311`	`311`	`image_slice = image[0, -3:, -3:, -1].flatten()`
`312`	`312`	`assert image.shape == (1, 512, 512, 3)`
`313`	`313`	`expected_slice = np.array(`
`314`		`- [0.41597816, 0.39302617, 0.44287828, 0.2687074, 0.28315824, 0.40582314, 0.20877528, 0.2380802, 0.39447647]`
	`314`	`+ [0.3876953, 0.40356445, 0.4934082, 0.39697266, 0.41674805, 0.41015625, 0.375, 0.36914062, 0.40649414]`
`315`	`315`	`)`
`316`	`316`	`assert (`
`317`	`317`	`np.abs(image_slice.flatten() - expected_slice).max() < 1e-3`