diff --git a/examples/parakeet-cli/README.md b/examples/parakeet-cli/README.md index 25a3f1e4..dfe58f57 100644 --- a/examples/parakeet-cli/README.md +++ b/examples/parakeet-cli/README.md @@ -30,7 +30,7 @@ options: -f, --file FILE [ ] input audio file -ng, --no-gpu [false ] disable GPU -dev N, --device N [0 ] GPU device to use - -fa, --flash-attn [true ] enable flash attention + -fa, --flash-attn [false ] enable flash attention -nfa, --no-flash-attn [false ] disable flash attention -ps, --print-segments [false ] print segment information ``` diff --git a/examples/parakeet-cli/parakeet-cli.cpp b/examples/parakeet-cli/parakeet-cli.cpp index bdb1739c..35f23819 100644 --- a/examples/parakeet-cli/parakeet-cli.cpp +++ b/examples/parakeet-cli/parakeet-cli.cpp @@ -62,7 +62,7 @@ static bool parakeet_params_parse(int argc, char ** argv, parakeet_params & para else if (arg == "-f" || arg == "--file") { params.fname_inp.emplace_back(ARGV_NEXT); } else if (arg == "-ng" || arg == "--no-gpu") { params.use_gpu = false; } else if (arg == "-dev" || arg == "--device") { params.gpu_device = std::stoi(ARGV_NEXT); } - else if (arg == "-fa" || arg == "--flash-attn") { params.flash_attn = true; } + else if (arg == "-fa" || arg == "--flash-attn") { params.flash_attn = false; } else if (arg == "-nfa" || arg == "--no-flash-attn") { params.flash_attn = false; } else if (arg == "-ps" || arg == "--print-segments") { params.print_segments = true; } else {