Fix log

Author: EricLBuehler

mistralrs-paged-attn/src/cuda/backend/paged_attention.rs

@@ -142,7 +142,7 @@ impl PagedAttention {
             candle::bail!(
                 "shape mismatch value_cache {:?}, expected {:?}",
                 vc_l.shape(),
-                (num_blocks, num_heads, head_size / x, block_size, x)
+                (num_blocks, num_kv_heads, head_size / x, block_size, x)
             )
         }
 

mistralrs-paged-attn/src/metal/backend/paged_attention.rs

@@ -128,7 +128,7 @@ impl candle_core::CustomOp1 for PagedAttention {
             candle_core::bail!(
                 "shape mismatch value_cache {:?}, expected {:?}",
                 vc_l.shape(),
-                (num_blocks, num_heads, head_size / x, block_size, x)
+                (num_blocks, num_kv_heads, head_size / x, block_size, x)
             )
         }