Fix bug: Incorrect variable used in rem_total_token_offset calculatio… (#13201)
This commit is contained in:
@@ -704,7 +704,7 @@ class PrefillAdder:
|
||||
for i, running_req in enumerate(self.running_batch.reqs):
|
||||
if running_req in preemptible_reqs:
|
||||
self.rem_total_token_offset -= (
|
||||
self._get_running_request_total_token_offset(req)
|
||||
self._get_running_request_total_token_offset(running_req)
|
||||
)
|
||||
release_counter += 1
|
||||
self.running_batch.release_req(
|
||||
|
||||
Reference in New Issue
Block a user