We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 8834aaf commit 2f4d576Copy full SHA for 2f4d576
backends/neuron/tests/server/test_prefill.py
@@ -56,6 +56,7 @@ def _test_prefill(config_name, generator, batch_size, do_sample):
56
assert tokens.ids[0] == expectations[0]
57
assert tokens.texts[0] == expectations[1]
58
59
+
60
def test_prefill_truncate(neuron_model_config):
61
config_name = neuron_model_config["name"]
62
neuron_model_path = neuron_model_config["neuron_model_path"]
0 commit comments