makaveli10 commited on
Commit
a291214
·
1 Parent(s): 90a66d9

add model_name to whisper tensorrt build command

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -29,7 +29,7 @@ wget --directory-prefix=assets https://openaipublic.azureedge.net/main/whisper/m
29
  - Build `small.en` TensorRT engine.
30
  ```bash
31
  pip install -r requirements.txt
32
- python3 build.py --output_dir whisper_small_en --use_gpt_attention_plugin --use_gemm_plugin --use_layernorm_plugin --use_bert_attention_plugin
33
  ```
34
 
35
  ### Mistral
 
29
  - Build `small.en` TensorRT engine.
30
  ```bash
31
  pip install -r requirements.txt
32
+ python3 build.py --output_dir whisper_small_en --use_gpt_attention_plugin --use_gemm_plugin --use_layernorm_plugin --use_bert_attention_plugin --model_name small.en
33
  ```
34
 
35
  ### Mistral