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

neuron: black

Browse Source
This commit is contained in:
Daniël de Kok 2025-08-28 14:29:38 +00:00
parent 8834aaf46d
commit 2f4d576a4c
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
backends/neuron/tests/server/test_prefill.py
View File

@ -56,6 +56,7 @@ def _test_prefill(config_name, generator, batch_size, do_sample):
assert tokens.ids[0] == expectations[0]
assert tokens.texts[0] == expectations[1]
def test_prefill_truncate(neuron_model_config):
config_name = neuron_model_config["name"]
neuron_model_path = neuron_model_config["neuron_model_path"]