diff --git a/common/common.cpp b/common/common.cpp index 9b313479..d0f38ed3 100644 --- a/common/common.cpp +++ b/common/common.cpp @@ -1152,7 +1152,7 @@ bool gpt_params_find_arg(int argc, char ** argv, const std::string & arg, gpt_pa return true; } if (arg == "--cpu-moe" || arg == "-cmoe") { - params.tensor_buft_overrides.push_back({strdup("\\.ffn_(up|down|gate)_exps"), ggml_backend_cpu_buffer_type()}); + params.tensor_buft_overrides.push_back({strdup("\\.ffn_(up|down|gate)_exps\\.weight"), ggml_backend_cpu_buffer_type()}); return true; } if (arg == "--n-cpu-moe" || arg == "-ncmoe") {