update note

Signed-off-by: Chen Zhang <[email protected]>
vllm-project · Jan 29, 2025 · e1399a8 · e1399a8
1 parent 92d3397
commit e1399a8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/vllm/attention/layer.py b/vllm/attention/layer.py
@@ -149,8 +149,8 @@ def forward(
         attn_metadata: AttentionMetadata,
     ) -> torch.Tensor:
         # NOTE: please avoid accessing `kv_cache` and `attn_metadata` arguments
-        # directly. Instead, please use the `self.kv_cache` and
-        # `get_forward_context().attn_metadata` to access them.
+        # directly, use `self.kv_cache` and
+        # `get_forward_context().attn_metadata` instead.
         if self.calculate_kv_scales:
             ctx_attn_metadata = get_forward_context().attn_metadata
             if ctx_attn_metadata.enable_kv_scales_calculation: