Add explanation comment

Signed-off-by: Tomer Asida <[email protected]>

Author: tomeras91

tensorrt_llm/_torch/models/checkpoints/hf/weight_mapper.py

@@ -59,6 +59,7 @@ def should_skip_module(self, module_name: str) -> bool:
     def _duplicate_kv_weights(self, module: nn.Module, new_name: str,
                               weights: dict):
         if new_name in ['k_proj', 'v_proj']:
+            # k_proj and v_proj shape is [num_kv_heads*head_dim, hidden_dim]
             num_kv_heads = weights['weight'].shape[0] // self._head_dim
             processed_weights = {
                 k: