node : add flash_attn param (#2170)

This commit is contained in:
Pedro Probst
2024-05-20 03:08:48 -03:00
committed by GitHub
parent 4798be1f9a
commit adee3f9c1f
3 changed files with 6 additions and 0 deletions

View File

@ -12,6 +12,7 @@ const whisperParams = {
model: path.join(__dirname, "../../models/ggml-base.en.bin"),
fname_inp: path.join(__dirname, "../../samples/jfk.wav"),
use_gpu: true,
flash_attn: false,
no_prints: true,
comma_in_time: false,
translate: true,