remove func

grimoire · grimoire · commit 7282a8470916 · 2025-05-14T16:32:17.000+08:00
diff --git a/lmdeploy/pytorch/engine/model_agent.py b/lmdeploy/pytorch/engine/model_agent.py
@@ -406,11 +406,6 @@ def __update_inputs(next_token_ids, model_metas):
             if sampling_inputs.random_offsets is not None:
                 sampling_inputs.random_offsets += 1
 
-        async def __await_distworker(worker, timeout: float = 0.001):
-            while not worker.is_completed():
-                await asyncio.sleep(timeout)
-            worker.wait()
-
         def __try_copy_everything():
             """try copy all inputs."""
             nonlocal inputs, all_ids, guided_input_ids, sampling_inputs, num_appendable_ids, num_ignore_eos
