fix: remove ggml_repeat

This commit is contained in:
namtranase 2023-12-22 22:45:35 +07:00
parent e8fae2db7a
commit a600c61da2

View file

@ -4072,9 +4072,7 @@ static struct ggml_tensor * llm_build_ffn_mpt_awq(
{
cur = ggml_gelu(ctx, cur);
cb(cur, "ffn_relu", il);
struct ggml_tensor *repeat = ggml_repeat(ctx, act_scales, cur);
cb(repeat, "ffn_repeat(scales)", il);
cur = ggml_div(ctx, cur, repeat);
cur = ggml_div(ctx, cur, act_scales);
cb(cur, "ffn_div(gelu)", il);
} break;
}