check

huggingface · Feb 7, 2024 · 90653b0 · 90653b0
1 parent 582c47a
commit 90653b0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/models/llama/test_inference_llama.py b/tests/models/llama/test_inference_llama.py
@@ -33,6 +33,6 @@ def test_foo(self):
             generate_ids = model.generate(inputs.input_ids, max_length=30)
         output = tokenizer.batch_decode(generate_ids, skip_special_tokens=True, clean_up_tokenization_spaces=False)[0]
 
-        expected_output = "Hey, are you conscious? Can you talk to me?\nI'm here, but I'm not sure I'm conscious."
+        expected_output = "Hey, are you conscious? Can you talk to me?\nI'm not sure if you can hear me, but I'm talking"
 
         assert output == expected_output