From c0c5acff077ac7c8fe68a0fdbad24306dbd9d4e3 Mon Sep 17 00:00:00 2001 From: ivarflakstad <69173633+ivarflakstad@users.noreply.github.com> Date: Tue, 4 Mar 2025 15:24:27 +0100 Subject: [PATCH] Fix bamba tests amd (#36535) --- tests/models/bamba/test_modeling_bamba.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/tests/models/bamba/test_modeling_bamba.py b/tests/models/bamba/test_modeling_bamba.py index aa5bbbfba6..3cbc239a5e 100644 --- a/tests/models/bamba/test_modeling_bamba.py +++ b/tests/models/bamba/test_modeling_bamba.py @@ -510,7 +510,7 @@ class BambaModelIntegrationTest(unittest.TestCase): EXPECTED_TEXTS = { # 7: "", 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) @@ -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|>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)