Skip to content

Commit 93391da

Browse files
.
1 parent 0b30139 commit 93391da

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

examples/whisper/README.md

+2-2
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ trtllm-build --checkpoint_dir ${checkpoint_dir}/decoder \
8787
--gemm_plugin ${INFERENCE_PRECISION} \
8888
--bert_attention_plugin ${INFERENCE_PRECISION} \
8989
--gpt_attention_plugin ${INFERENCE_PRECISION} \
90-
--remove_input_padding disable
90+
--remove_input_padding enable
9191
```
9292

9393
### Run
@@ -121,7 +121,7 @@ WEIGHT_ONLY_PRECISION=int8
121121
MAX_BEAM_WIDTH=4
122122
MAX_BATCH_SIZE=8
123123
checkpoint_dir=distil_whisper_medium_en_weights_${WEIGHT_ONLY_PRECISION}
124-
output_dir=distil_whisper_medium_en${WEIGHT_ONLY_PRECISION}
124+
output_dir=distil_whisper_medium_en_${WEIGHT_ONLY_PRECISION}
125125

126126
python3 convert_checkpoint.py \
127127
--use_weight_only \

0 commit comments

Comments
 (0)