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

Fix bamba tests amd (#36535)

Browse Source
This commit is contained in:
ivarflakstad
2025-03-04 15:24:27 +01:00
committed by GitHub
parent 37508816d6
commit c0c5acff07
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tests/models/bamba/test_modeling_bamba.py
View File

@@ -510,7 +510,7 @@ class BambaModelIntegrationTest(unittest.TestCase):
EXPECTED_TEXTS = { EXPECTED_TEXTS = {
# 7: "", # 7: "",
8: "<|begin_of_text|>Hey how are you doing on this lovely evening? I hope you are all having a good time.", 8: "<|begin_of_text|>Hey how are you doing on this lovely evening? I hope you are all having a good time.",
# 9: """, 9: "<|begin_of_text|>Hey how are you doing on this lovely evening? I hope you are doing well. I am here",
} }
self.model.to(torch_device) self.model.to(torch_device)
@@ -549,7 +549,10 @@ class BambaModelIntegrationTest(unittest.TestCase):
"<|begin_of_text|>Hey how are you doing on this lovely evening? I hope you are doing well. I am here", "<|begin_of_text|>Hey how are you doing on this lovely evening? I hope you are doing well. I am here",
"!!!<|begin_of_text|>I am late! I need to get to work! I have to get to the", "!!!<|begin_of_text|>I am late! I need to get to work! I have to get to the",
], ],
9: [], 9: [
"<|begin_of_text|>Hey how are you doing on this lovely evening? I hope you are doing well. I am here",
"!!!<|begin_of_text|>I am late! I need to be at the airport in 20 minutes! I",
],
} }
self.model.to(torch_device) self.model.to(torch_device)