Skip to content

Commit 393c8f5

Browse files
authored
[hotfix] fix inference typo (#5438)
1 parent 785cd9a commit 393c8f5

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

colossalai/legacy/inference/async_manager.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -55,14 +55,14 @@ def _step(self):
5555
self.stats_tool.count_prompt_tokens(new_batch)
5656
self.running_batch = new_batch
5757
has_new_finished, outputs = self._prefill_batch(self.running_batch)
58-
self._filter_runing_batch()
58+
self._filter_running_batch()
5959
self.has_wait_tokens = 0
6060

6161
else:
6262
if self.has_wait_tokens < self.max_wait_tokens:
6363
self.stats_tool.count_output_tokens(self.running_batch)
6464
has_new_finished, outputs = self._decode_batch(self.running_batch)
65-
self._filter_runing_batch()
65+
self._filter_running_batch()
6666
self.has_wait_tokens += 1
6767

6868
else:
@@ -78,7 +78,7 @@ def _step(self):
7878
else:
7979
self.stats_tool.count_output_tokens(self.running_batch)
8080
has_new_finished, outputs = self._decode_batch(self.running_batch)
81-
self._filter_runing_batch()
81+
self._filter_running_batch()
8282
self.has_wait_tokens += 1
8383

8484
if has_new_finished:

colossalai/legacy/inference/manager.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -131,14 +131,14 @@ def _step(self):
131131
self.stats_tool.count_prompt_tokens(new_batch)
132132
self.running_batch = new_batch
133133
yield from self._prefill_batch(self.running_batch)
134-
self._filter_runing_batch()
134+
self._filter_running_batch()
135135
self.has_wait_tokens = 0
136136
return
137137

138138
if self.has_wait_tokens < self.max_wait_tokens:
139139
self.stats_tool.count_output_tokens(self.running_batch)
140140
yield from self._decode_batch(self.running_batch)
141-
self._filter_runing_batch()
141+
self._filter_running_batch()
142142
self.has_wait_tokens += 1
143143
return
144144
else:
@@ -154,7 +154,7 @@ def _step(self):
154154
else:
155155
self.stats_tool.count_output_tokens(self.running_batch)
156156
yield from self._decode_batch(self.running_batch)
157-
self._filter_runing_batch()
157+
self._filter_running_batch()
158158
self.has_wait_tokens += 1
159159

160160
return
@@ -243,7 +243,7 @@ def _handle_finish_req(self, batch: Batch, has_new_finished_req):
243243
self._filter_batch(batch)
244244
yield from self._output_process(finished_reqs)
245245

246-
def _filter_runing_batch(self):
246+
def _filter_running_batch(self):
247247
if self.running_batch is not None and self.running_batch.is_clear():
248248
self.running_batch = None
249249

0 commit comments

Comments
 (0)