Explorar o código

k-quants : fix zero-weight guard in Q6_K (ref #3040)

Georgi Gerganov %!s(int64=2) %!d(string=hai) anos
pai
achega
178b1850eb
Modificáronse 1 ficheiros con 1 adicións e 0 borrados
  1. 1 0
      k_quants.c

+ 1 - 0
k_quants.c

@@ -1089,6 +1089,7 @@ void quantize_row_q6_K_reference(const float * restrict x, block_q6_K * restrict
         if (!max_abs_scale) {
             memset(&y[i], 0, sizeof(block_q6_K));
             y[i].d = ggml_fp32_to_fp16(0.f);
+            x += QK_K;
             continue;
         }