@@ -131,14 +131,14 @@ def _step(self):
131
131
self .stats_tool .count_prompt_tokens (new_batch )
132
132
self .running_batch = new_batch
133
133
yield from self ._prefill_batch (self .running_batch )
134
- self ._filter_runing_batch ()
134
+ self ._filter_running_batch ()
135
135
self .has_wait_tokens = 0
136
136
return
137
137
138
138
if self .has_wait_tokens < self .max_wait_tokens :
139
139
self .stats_tool .count_output_tokens (self .running_batch )
140
140
yield from self ._decode_batch (self .running_batch )
141
- self ._filter_runing_batch ()
141
+ self ._filter_running_batch ()
142
142
self .has_wait_tokens += 1
143
143
return
144
144
else :
@@ -154,7 +154,7 @@ def _step(self):
154
154
else :
155
155
self .stats_tool .count_output_tokens (self .running_batch )
156
156
yield from self ._decode_batch (self .running_batch )
157
- self ._filter_runing_batch ()
157
+ self ._filter_running_batch ()
158
158
self .has_wait_tokens += 1
159
159
160
160
return
@@ -243,7 +243,7 @@ def _handle_finish_req(self, batch: Batch, has_new_finished_req):
243
243
self ._filter_batch (batch )
244
244
yield from self ._output_process (finished_reqs )
245
245
246
- def _filter_runing_batch (self ):
246
+ def _filter_running_batch (self ):
247
247
if self .running_batch is not None and self .running_batch .is_clear ():
248
248
self .running_batch = None
249
249
0 commit comments