Skip to content

Commit be1a8cb

Browse files
author
Iwan Kawrakow
committed
Fix flash attention long argument for mainloine compatibility
1 parent f4202c8 commit be1a8cb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

common/common.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1135,7 +1135,7 @@ bool gpt_params_find_arg(int argc, char ** argv, const std::string & arg, gpt_pa
11351135
params.flash_attn = false;
11361136
return true;
11371137
}
1138-
if (arg == "-fa" || arg == "--flash-attention") {
1138+
if (arg == "-fa" || arg == "--flash-attn") {
11391139
CHECK_ARG
11401140
std::string next_arg{argv[i]};
11411141
for (auto& c : next_arg) c = std::tolower(c);

0 commit comments

Comments
 (0)