yuekai
/

whisper_turbo_and_distill_tllm_checkpoint

Model card Files Files and versions Community

yuekai commited on Oct 18

Commit

343e50d

•

1 Parent(s): 7e28c75

Update build.sh

Files changed (1) hide show

build.sh +2 -2

build.sh CHANGED Viewed

@@ -4,8 +4,8 @@ INFERENCE_PRECISION=float16
 MAX_BEAM_WIDTH=4
 MAX_BATCH_SIZE=64
-checkpoint_dir=$1
-output_dir=$2
 trtllm-build --checkpoint_dir ${checkpoint_dir}/encoder \
               --output_dir ${output_dir}/encoder \

 MAX_BEAM_WIDTH=4
 MAX_BATCH_SIZE=64
+checkpoint_dir=tllm_checkpoint_turbo
+output_dir=whisper_turbo
 trtllm-build --checkpoint_dir ${checkpoint_dir}/encoder \
               --output_dir ${output_dir}/encoder \