Procházet zdrojové kódy

ggml-metal: fix yarn rope (#3937)

Xiao-Yong Jin před 2 roky
rodič
revize
5ba3746171
1 změnil soubory, kde provedl 2 přidání a 1 odebrání
  1. 2 1
      ggml-metal.m

+ 2 - 1
ggml-metal.m

@@ -1403,7 +1403,8 @@ void ggml_metal_graph_compute(
                             const int n_past     = ((int32_t *) dst->op_params)[0];
                             const int n_past     = ((int32_t *) dst->op_params)[0];
                             const int n_dims     = ((int32_t *) dst->op_params)[1];
                             const int n_dims     = ((int32_t *) dst->op_params)[1];
                             const int mode       = ((int32_t *) dst->op_params)[2];
                             const int mode       = ((int32_t *) dst->op_params)[2];
-                            const int n_orig_ctx = ((int32_t *) dst->op_params)[3];
+                            // skip 3, n_ctx, used in GLM RoPE, unimplemented in metal
+                            const int n_orig_ctx = ((int32_t *) dst->op_params)[4];
 
 
                             float freq_base, freq_scale, ext_factor, attn_factor, beta_fast, beta_slow;
                             float freq_base, freq_scale, ext_factor, attn_factor, beta_fast, beta_slow;
                             memcpy(&freq_base,   (int32_t *) dst->op_params +  5, sizeof(float));
                             memcpy(&freq_base,   (int32_t *) dst->op_params +  5, sizeof(float));