Views
No views yet
1# Install TensorRT-LLM
2# See: https://github.com/NVIDIA/TensorRT-LLM
3
4# Clone TensorRT-LLM repository
5git clone https://github.com/NVIDIA/TensorRT-LLM.git
6cd TensorRT-LLM/examples/whisper
7
8# Build the encoder engine
9trtllm-build --checkpoint_dir /path/to/this/repo/encoder \
10 --output_dir /path/to/output/encoder \
11 --moe_plugin disable \
12 --max_batch_size 8 \
13 --gemm_plugin disable \
14 --bert_attention_plugin float16 \
15 --max_input_len 3000 --max_seq_len=3000
16
17# Build the decoder engine
18trtllm-build --checkpoint_dir /path/to/this/repo/decoder \
19 --output_dir /path/to/output/decoder \
20 --moe_plugin disable \
21 --max_beam_width 4 \
22 --max_batch_size 8 \
23 --max_seq_len 114 \
24 --max_input_len 14 \
25 --max_encoder_input_len 3000 \
26 --gemm_plugin float16 \
27 --bert_attention_plugin float16 \
28 --gpt_attention_plugin float16