File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -113,11 +113,11 @@ class ChunkedContextMetadata:
113113 chunk_seq_lens_npu : torch .Tensor
114114 # for mla DCP & PCP
115115 padded_chunk_seq_lens_npu : torch .Tensor = None
116- padded_local_chunk_seq_lens : Optional [list [list [int ]]] = None
117- local_context_lens_allranks : Optional [list [list [int ]]] = None
116+ padded_local_chunk_seq_lens : Optional [list [list [int ]]] = None
117+ local_context_lens_allranks : Optional [list [list [int ]]] = None
118118 padded_local_cu_seq_lens : torch .Tensor = None
119- cu_seq_lens_lst : Optional [list [list [int ]]] = None
120- chunk_size : Optional [int ] = None
119+ cu_seq_lens_lst : Optional [list [list [int ]]] = None
120+ chunk_size : Optional [int ] = None
121121
122122 attn_mask : torch .Tensor
123123 query_lens : torch .Tensor
You can’t perform that action at this time.
0 commit comments