huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-09 11:24:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

test(neuron): adjust expectations

Browse Source 
Since the latest optimum-neuron uses a new modeling for granite and
qwen, the greedy outputs are slighly different.
This commit is contained in:
David Corvoysier 2025-06-27 12:31:31 +00:00
parent c45977e48b
commit 83b766f9ef
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
integration-tests/neuron/test_generate.py
View File

@ -21,8 +21,8 @@ async def test_model_single_request(tgi_service):
assert response.details.generated_tokens == 17 assert response.details.generated_tokens == 17
greedy_expectations = { greedy_expectations = {
"llama": " and how does it work?\nDeep learning is a subset of machine learning that uses artificial", "llama": " and how does it work?\nDeep learning is a subset of machine learning that uses artificial",
"qwen2": " - Part 1\n\nDeep Learning is a subset of Machine Learning that is based on", "qwen2": " - Deep Learning is a subset of Machine Learning that involves the use of artificial neural networks",
"granite": "\n\nDeep Learning is a subset of Machine Learning, which is a branch of Art", "granite": "\n\nDeep learning is a subset of machine learning techniques based on artificial neural networks",
} }
assert response.generated_text == greedy_expectations[service_name] assert response.generated_text == greedy_expectations[service_name]
@ -78,8 +78,8 @@ async def test_model_multiple_requests(tgi_service, neuron_generate_load):
assert len(responses) == 4 assert len(responses) == 4
expectations = { expectations = {
"llama": "Deep learning is a subset of machine learning that uses artificial", "llama": "Deep learning is a subset of machine learning that uses artificial",
"qwen2": "Deep Learning is a subset of Machine Learning that is based on", "qwen2": "Deep Learning is a subset of Machine Learning that involves",
"granite": "Deep Learning is a subset of Machine Learning, which is a branch of Art", "granite": "Deep learning is a subset of machine learning techniques",
} }
expected = expectations[tgi_service.client.service_name] expected = expectations[tgi_service.client.service_name]
for r in responses: for r in responses: