Skip to content

Commit 898c8e7

Browse files
committed
Revert "llama: -fa 1/0/-1 aliases for -fa on/off/auto (ggml-org#15746)"
This reverts commit c466abe.
1 parent bb15126 commit 898c8e7

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

common/arg.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1550,11 +1550,11 @@ common_params_context common_params_parser_init(common_params & params, llama_ex
15501550
{"-fa", "--flash-attn"}, "FA",
15511551
string_format("set Flash Attention use ('on', 'off', or 'auto', default: '%s')", llama_flash_attn_type_name(params.flash_attn_type)),
15521552
[](common_params & params, const std::string & value) {
1553-
if (value == "on" || value == "enabled" || value == "1") {
1553+
if (value == "on" || value == "enabled") {
15541554
params.flash_attn_type = LLAMA_FLASH_ATTN_TYPE_ENABLED;
1555-
} else if (value == "off" || value == "disabled" || value == "0") {
1555+
} else if (value == "off" || value == "disabled") {
15561556
params.flash_attn_type = LLAMA_FLASH_ATTN_TYPE_DISABLED;
1557-
} else if (value == "auto" || value == "-1") {
1557+
} else if (value == "auto") {
15581558
params.flash_attn_type = LLAMA_FLASH_ATTN_TYPE_AUTO;
15591559
} else {
15601560
throw std::runtime_error(string_format("error: unkown value for --flash-attn: '%s'\n", value.c_str()));

0 commit comments

Comments
 (0)