File tree Expand file tree Collapse file tree 3 files changed +2
-5
lines changed Expand file tree Collapse file tree 3 files changed +2
-5
lines changed Original file line number Diff line number Diff line change @@ -85,8 +85,6 @@ class ForwardMeta:
85
85
# The sequence length processed in the current step
86
86
seq_lens_this_time : Optional [paddle .Tensor ] = None
87
87
88
- # Accumulated offset
89
- cum_offsets : Optional [paddle .Tensor ] = None
90
88
# batch_id_per_token tensor, used to indicate which token belongs which batch after padding removal to the original input_ids
91
89
batch_id_per_token : Optional [paddle .Tensor ] = None
92
90
# Accumulated sequence length of query
@@ -112,7 +110,8 @@ class XPUForwardMeta(ForwardMeta):
112
110
"""
113
111
XPUForwardMeta is used to store the global meta information of the forward, and some XPU specific meta info.
114
112
"""
115
-
113
+ # Accumulated offset
114
+ cum_offsets : Optional [paddle .Tensor ] = None
116
115
# TODO(wanghaitao): Supplementary notes
117
116
#
118
117
encoder_batch_map : Optional [paddle .Tensor ] = None
Original file line number Diff line number Diff line change @@ -397,7 +397,6 @@ def _initialize_forward_meta(self):
397
397
seq_lens_encoder = self .model_inputs ["seq_lens_encoder" ],
398
398
seq_lens_decoder = self .model_inputs ["seq_lens_decoder" ],
399
399
seq_lens_this_time = self .model_inputs ["seq_lens_this_time" ],
400
- cum_offsets = self .model_inputs ["cum_offsets" ],
401
400
batch_id_per_token = self .model_inputs ["batch_id_per_token" ],
402
401
cu_seqlens_q = self .model_inputs ["cu_seqlens_q" ],
403
402
cu_seqlens_k = self .model_inputs ["cu_seqlens_k" ],
Original file line number Diff line number Diff line change @@ -680,7 +680,6 @@ def initialize_forward_meta(self):
680
680
seq_lens_encoder = self .share_inputs ["seq_lens_encoder" ],
681
681
seq_lens_decoder = self .share_inputs ["seq_lens_decoder" ],
682
682
seq_lens_this_time = self .share_inputs ["seq_lens_this_time" ],
683
- cum_offsets = self .share_inputs ["cum_offsets" ],
684
683
batch_id_per_token = self .share_inputs ["batch_id_per_token" ],
685
684
cu_seqlens_q = self .share_inputs ["cu_seqlens_q" ],
686
685
cu_seqlens_k = self .share_inputs ["cu_seqlens_k" ],
You can’t perform that action at this time.
0 commit comments