SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Add AMD expectations to Mistral3 tests (#39481)

Browse Source 
Add AMD expectations to mistral3 tests
This commit is contained in:
Ákos Hadnagy
2025-07-22 15:40:16 +02:00
committed by GitHub
parent 1806583390
commit b62557e712
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/mistral3/test_modeling_mistral3.py
View File

@@ -317,6 +317,7 @@ class Mistral3IntegrationTest(unittest.TestCase):
{ {
("xpu", 3): "The image features two cats resting on a pink blanket. The cat on the left is a kitten", ("xpu", 3): "The image features two cats resting on a pink blanket. The cat on the left is a kitten",
("cuda", 8): 'The image features two cats lying on a pink surface, which appears to be a couch or a bed', ("cuda", 8): 'The image features two cats lying on a pink surface, which appears to be a couch or a bed',
("rocm", (9, 5)): "The image features two tabby cats lying on a pink surface, which appears to be a cushion or"
} }
) # fmt: skip ) # fmt: skip
expected_output = expected_outputs.get_expectation() expected_output = expected_outputs.get_expectation()
@@ -363,6 +364,7 @@ class Mistral3IntegrationTest(unittest.TestCase):
{ {
("xpu", 3): "Calm lake's mirror gleams,\nWhispering pines stand in silence,\nPath to peace begins.", ("xpu", 3): "Calm lake's mirror gleams,\nWhispering pines stand in silence,\nPath to peace begins.",
("cuda", 8): "Wooden path to calm,\nReflections whisper secrets,\nNature's peace unfolds.", ("cuda", 8): "Wooden path to calm,\nReflections whisper secrets,\nNature's peace unfolds.",
("rocm", (9, 5)): "Calm waters reflect\nWooden path to distant shore\nSilence in the scene"
} }
) # fmt: skip ) # fmt: skip
expected_output = expected_outputs.get_expectation() expected_output = expected_outputs.get_expectation()