better description of indices variable in comment

Signed-off-by: Yannick Schnider <Yannick.Schnider1@ibm.com>
IBM · Jan 31, 2025 · fc09267 · fc09267
1 parent 3f087a7
commit fc09267
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/vllm/model_executor/model_loader/spyre.py b/vllm/model_executor/model_loader/spyre.py
@@ -46,7 +46,9 @@ def __init__(
         self.past_key_value_states = None
         self.dtype = torch.float16 if envs.VLLM_SPYRE_DYNAMO_BACKEND == \
             'sendnn_decoder' else torch.float32
-        # indices: True unfinished, False for finished or padded sequence
+        # boolean tensor of length batch size with indices:
+        # True for unfinished sequences and
+        # False for finished or padded sequences
         self.indices = None
 
         # Lazy initialized