Remove overlap thread (#11210)
Co-authored-by: Lianmin Zheng <15100009+merrymercy@users.noreply.github.com> Co-authored-by: Hanming Lu <69857889+hanming-lu@users.noreply.github.com>
This commit is contained in:
co-authored by
Lianmin Zheng
Hanming Lu
parent
24bc3fb0f9
commit
1519a89cfd
@@ -37,8 +37,7 @@ class FutureMap:
|
||||
return cur_future_ct
|
||||
|
||||
def resolve_future(self, model_worker_batch: ModelWorkerBatch):
|
||||
input_ids = model_worker_batch.input_ids
|
||||
_resolve_future_token_ids(input_ids, self.token_ids_buf)
|
||||
_resolve_future_token_ids(model_worker_batch.input_ids, self.token_ids_buf)
|
||||
|
||||
def update_next_future(self, future_ct: int, bs: int):
|
||||
return torch.arange(
|
||||
|
||||
Reference in New Issue
Block a user