Skip to content

Commit cb0518a

Browse files
committed
no canny
1 parent 5b49149 commit cb0518a

File tree

1 file changed

+1
-28
lines changed

1 file changed

+1
-28
lines changed

tests/pipelines/controlnet_hunyuandit/test_controlnet_hunyuandit.py

Lines changed: 1 addition & 28 deletions
Original file line numberDiff line numberDiff line change
@@ -238,34 +238,7 @@ def test_canny(self):
238238

239239
original_image = image[-3:, -3:, -1].flatten()
240240

241-
if torch_device == "xpu":
242-
expected_image = np.array(
243-
[
244-
0.2944336,
245-
0.30981445,
246-
0.24389648,
247-
0.2890625,
248-
0.32006836,
249-
0.2578125,
250-
0.31469727,
251-
0.34545898,
252-
0.28515625,
253-
]
254-
)
255-
else:
256-
expected_image = np.array(
257-
[
258-
0.43652344,
259-
0.4399414,
260-
0.44921875,
261-
0.45043945,
262-
0.45703125,
263-
0.44873047,
264-
0.43579102,
265-
0.44018555,
266-
0.42578125,
267-
]
268-
)
241+
expected_image = np.array([0.43652344, 0.4399414, 0.44921875, 0.45043945, 0.45703125, 0.44873047, 0.43579102, 0.44018555, 0.42578125])
269242

270243
assert np.abs(original_image.flatten() - expected_image).max() < 1e-2
271244

0 commit comments

Comments
 (0)