Skip to content

Commit 76ebf07

Browse files
fix some more failures on nvidia
1 parent 01f720a commit 76ebf07

File tree

3 files changed

+6
-9
lines changed

3 files changed

+6
-9
lines changed

tests/models/qwen2/test_modeling_qwen2.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -210,11 +210,8 @@ def test_export_static_cache(self):
210210
tokenizer = AutoTokenizer.from_pretrained(qwen_model, pad_token="</s>", padding_side="right")
211211

212212
expected_text_completions = Expectations({
213-
("cuda", None): [
214-
"My favourite condiment is 100% natural, organic, gluten free, vegan, and free from preservatives. I"
215-
],
216213
("cuda", 8): [
217-
"My favourite condiment is 100% natural, organic, gluten free, vegan, and vegetarian. I love to use"
214+
"My favourite condiment is 100% natural, organic, gluten free, vegan, and free from preservatives. I"
218215
],
219216
("rocm", (9, 4)): [
220217
"My favourite condiment is 100% natural, organic and vegan. I love to use it in my cooking, but"

tests/models/qwen2_5_omni/test_modeling_qwen2_5_omni.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -684,7 +684,7 @@ def test_small_model_integration_test(self):
684684

685685
EXPECTED_DECODED_TEXT = Expectations({
686686
("xpu", None): "system\nYou are a helpful assistant.\nuser\nWhat's that sound and what kind of dog is this?\nassistant\nThe sound is glass shattering, and the dog is a Labrador Retriever.",
687-
("cuda", (8, 6)): "system\nYou are a helpful assistant.\nuser\nWhat's that sound and what kind of dog is this?\nassistant\nThe sound is glass shattering, and the dog is a Labrador Retriever.",
687+
("cuda", (8, 6)): "system\nYou are a helpful assistant.\nuser\nWhat's that sound and what kind of dog is this?\nassistant\nThe sound is a glass shattering. The dog in the picture is a Labrador Retriever.",
688688
("rocm", (9, 4)): "system\nYou are a helpful assistant.\nuser\nWhat's that sound and what kind of dog is this?\nassistant\nThe sound is glass shattering, and the dog is a Labrador Retriever.",
689689
}).get_expectation() # fmt: skip
690690

@@ -720,8 +720,8 @@ def test_small_model_integration_test_batch(self):
720720
"system\nYou are a helpful assistant.\nuser\nWhat's that sound and what kind of dog is this?\nassistant\nThe sound is of glass shattering, and the dog in the picture is a Labrador Retriever",
721721
],
722722
("cuda", 8): [
723-
"system\nYou are a helpful assistant.\nuser\nWhat's that sound and what kind of dog is this?\nassistant\nThe sound is glass shattering, and the dog is a Labrador Retriever.",
724-
"system\nYou are a helpful assistant.\nuser\nWhat's that sound and what kind of dog is this?\nassistant\nThe sound is glass shattering, and the dog is a Labrador Retriever.",
723+
"system\nYou are a helpful assistant.\nuser\nWhat's that sound and what kind of dog is this?\nassistant\nThe sound is a glass shattering. The dog in the picture is a Labrador Retriever.",
724+
"system\nYou are a helpful assistant.\nuser\nWhat's that sound and what kind of dog is this?\nassistant\nThe sound is a glass shattering. The dog in the picture is a Labrador Retriever.",
725725
],
726726
("rocm", (9, 4)): [
727727
"system\nYou are a helpful assistant.\nuser\nWhat's that sound and what kind of dog is this?\nassistant\nThe sound is a glass shattering. The dog in the picture is a Labrador Retriever.",

tests/models/qwen2_5_vl/test_modeling_qwen2_5_vl.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -572,8 +572,8 @@ def test_small_model_integration_test_batch_different_resolutions(self):
572572
"system\nYou are a helpful assistant.\nuser\nWhat kind of dog is this?\nassistant\n addCriterion\nThe dog in the picture appears to be a Labrador Retriever. Labradors are known for their friendly and gentle nature, which is",
573573
],
574574
("cuda", (8, 6)): [
575-
'system\nYou are a helpful assistant.\nuser\nWhat kind of dog is this?\nassistant\nThe dog in the picture appears to be a Labrador Retriever. Labradors are known for their friendly and energetic nature, which is evident in',
576-
'system\nYou are a helpful assistant.\nuser\nWhat kind of dog is this?\nassistant\nThe dog in the picture appears to be a Labrador Retriever. Labradors are known for their friendly and energetic nature, which is evident in',
575+
'system\nYou are a helpful assistant.\nuser\nWhat kind of dog is this?\nassistant\n addCriterion\nThe dog in the picture appears to be a Labrador Retriever. Labradors are known for their friendly and gentle nature, which is',
576+
'system\nYou are a helpful assistant.\nuser\nWhat kind of dog is this?\nassistant\n addCriterion\nThe dog in the picture appears to be a Labrador Retriever. Labradors are known for their friendly and gentle nature, which is',
577577
],
578578
("rocm", None): [
579579
'system\nYou are a helpful assistant.\nuser\nWhat kind of dog is this?\nassistant\nThe dog in the picture appears to be a Labrador Retriever. Labradors are known for their friendly and energetic nature, which is evident in',

0 commit comments

Comments
 (0)