Skip to content

Commit 99010b4

Browse files
ahadnagyzaristei
authored andcommitted
Add AMD expectations to Mistral3 tests (huggingface#39481)
Add AMD expectations to mistral3 tests
1 parent a658a12 commit 99010b4

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/models/mistral3/test_modeling_mistral3.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -317,6 +317,7 @@ def test_mistral3_integration_generate(self):
317317
{
318318
("xpu", 3): "The image features two cats resting on a pink blanket. The cat on the left is a kitten",
319319
("cuda", 8): 'The image features two cats lying on a pink surface, which appears to be a couch or a bed',
320+
("rocm", (9, 5)): "The image features two tabby cats lying on a pink surface, which appears to be a cushion or"
320321
}
321322
) # fmt: skip
322323
expected_output = expected_outputs.get_expectation()
@@ -363,6 +364,7 @@ def test_mistral3_integration_batched_generate(self):
363364
{
364365
("xpu", 3): "Calm lake's mirror gleams,\nWhispering pines stand in silence,\nPath to peace begins.",
365366
("cuda", 8): "Wooden path to calm,\nReflections whisper secrets,\nNature's peace unfolds.",
367+
("rocm", (9, 5)): "Calm waters reflect\nWooden path to distant shore\nSilence in the scene"
366368
}
367369
) # fmt: skip
368370
expected_output = expected_outputs.get_expectation()

0 commit comments

Comments
 (0)