TensorRT-LLMs/tests/model/redrafter
Kaiyu Xie 2d234357c6
Update TensorRT-LLM (#1954)
* Update TensorRT-LLM

---------

Co-authored-by: Altair-Alpha <62340011+Altair-Alpha@users.noreply.github.com>
2024-07-16 15:30:25 +08:00
..
test_beams2tree.py Update TensorRT-LLM (#1954) 2024-07-16 15:30:25 +08:00
test_draft_token_indices.py Update TensorRT-LLM (#1954) 2024-07-16 15:30:25 +08:00
test_draft_token.py Update TensorRT-LLM (#1954) 2024-07-16 15:30:25 +08:00
test_gather_beams.py Update TensorRT-LLM (#1954) 2024-07-16 15:30:25 +08:00
test_mask.py Update TensorRT-LLM (#1954) 2024-07-16 15:30:25 +08:00
test_packed_position_ids.py Update TensorRT-LLM (#1954) 2024-07-16 15:30:25 +08:00
test_prefix_match_indices.py Update TensorRT-LLM (#1954) 2024-07-16 15:30:25 +08:00
test_prepare_input.py Update TensorRT-LLM (#1954) 2024-07-16 15:30:25 +08:00
test_process_logits.py Update TensorRT-LLM (#1954) 2024-07-16 15:30:25 +08:00
test_top1.py Update TensorRT-LLM (#1954) 2024-07-16 15:30:25 +08:00
test_unpack_gen_data.py Update TensorRT-LLM (#1954) 2024-07-16 15:30:25 +08:00
test_validate.py Update TensorRT-LLM (#1954) 2024-07-16 15:30:25 +08:00