Skip to content

Commit 020ace7

Browse files
authored
nomic_bert: make the test suitable for general device. (#45209)
Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com>
1 parent f6ff4ed commit 020ace7

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

tests/models/nomic_bert/test_modeling_nomic_bert.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -300,7 +300,7 @@ def test_inference_no_head_absolute_embedding_v1_5(self):
300300
# fmt: off
301301
expected_slice = Expectations(
302302
{
303-
("cuda", None): torch.tensor(
303+
(None, None): torch.tensor(
304304
[
305305
[
306306
[1.7039e00, -4.5610e00, 1.5236e00],
@@ -339,7 +339,7 @@ def test_inference_no_head_absolute_embedding_v1(self):
339339
# fmt: off
340340
expected_slice = Expectations(
341341
{
342-
("cuda", None): torch.tensor(
342+
(None, None): torch.tensor(
343343
[
344344
[
345345
[ 1.2961, -1.1757, 1.2094],
@@ -352,7 +352,7 @@ def test_inference_no_head_absolute_embedding_v1(self):
352352
[-0.4336, -0.8528, -0.2509],
353353
]
354354
]
355-
)
355+
),
356356
}
357357
).get_expectation()
358358
# fmt: on

0 commit comments

Comments
 (0)