Skip to content

Commit 9ea66a9

Browse files
committed
fix
1 parent 26d4d98 commit 9ea66a9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/models/phimoe/test_modeling_phimoe.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -211,6 +211,6 @@ def test_phimoe_instruct_with_static_cache(self):
211211
output_text = tokenizer.batch_decode(torch.tensor([response_tokens], dtype=torch.long, device=torch_device))
212212

213213
EXPECTED_OUTPUT = [
214-
"<|system|> You are a helpful digital assistant. Please provide safe, ethical and accurate information to the user.<|end|><|user|> Can you provide ways to eat combinations of bananas and dragonfruits?<|end|><|assistant|> awards"
214+
"<|system|> You are a helpful digital assistant. Please provide safe, ethical and accurate information to the user.<|end|><|user|> Can you provide ways to eat combinations of bananas and dragonfruits?<|end|><|assistant|> C"
215215
]
216216
self.assertListEqual(output_text, EXPECTED_OUTPUT)

0 commit comments

Comments
 (0)