examples : set flash attention to false for parakeet-cli [no ci]
This commit is contained in:
parent
1e0c5dd580
commit
1ed636a70d
|
|
@ -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
|
||||
```
|
||||
|
|
|
|||
|
|
@ -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 {
|
||||
|
|
|
|||
Loading…
Reference in New Issue