nicolas
/
ik_llama_opt
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix flash attention long argument for mainloine compatibility

This commit is contained in:
Kawrakow 2025-11-13 19:22:16 +02:00
parent ce3ce97a29
commit 22c20fcd6d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
common/common.cpp
View File

@ -1135,7 +1135,7 @@ bool gpt_params_find_arg(int argc, char ** argv, const std::string & arg, gpt_pa
params.flash_attn = false; params.flash_attn = false;
return true; return true;
} }
if (arg == "-fa" || arg == "--flash-attention") { if (arg == "-fa" || arg == "--flash-attn") {
CHECK_ARG CHECK_ARG
std::string next_arg{argv[i]}; std::string next_arg{argv[i]};
for (auto& c : next_arg) c = std::tolower(c); for (auto& c : next_arg) c = std::tolower(c);