Skip to content

Commit e451e5b

Browse files
committed
remove duplicate comments
1 parent 4a468e8 commit e451e5b

File tree

1 file changed

+0
-4
lines changed

1 file changed

+0
-4
lines changed

ucm/integration/vllm/ucm_connector.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -413,10 +413,6 @@ def _get_tensors(
413413
def _generate_task(
414414
self, vllm_block_ids: List[int], ucm_block_ids: List[bytes]
415415
) -> Tuple[List[bytes], List[int], List[List[torch.Tensor]]]:
416-
"""
417-
GQA/MHA: one layer shape is (2, num_blocks, block_size, num_kv_heads, head_size)
418-
MLA: one layer shape is (num_blocks, block_size, head_size)
419-
"""
420416
block_ids, shard_indexs, tensors = [], [], []
421417
for i, vllm_block_id in enumerate(vllm_block_ids):
422418
k_tensors, v_tensors = self._get_tensors(vllm_block_id)

0 commit comments

Comments
 (0)