File tree Expand file tree Collapse file tree 1 file changed +1
-1
lines changed Expand file tree Collapse file tree 1 file changed +1
-1
lines changed Original file line number Diff line number Diff line change @@ -50,6 +50,7 @@ def _init_executor(self) -> None:
50
50
self .is_failed = False
51
51
self .shutdown_event = threading .Event ()
52
52
self .failure_callback : Optional [FailureCallback ] = None
53
+ self .io_thread_pool : Optional [ThreadPoolExecutor ] = None
53
54
54
55
self .world_size = self .parallel_config .world_size
55
56
tensor_parallel_size = self .parallel_config .tensor_parallel_size
@@ -107,7 +108,6 @@ def _init_executor(self) -> None:
107
108
108
109
# For pipeline parallel, we use a thread pool for asynchronous
109
110
# execute_model.
110
- self .io_thread_pool : Optional [ThreadPoolExecutor ] = None
111
111
if self .max_concurrent_batches > 1 :
112
112
# Note: must use only 1 IO thread to keep dequeue sequence
113
113
# from the response queue
You can’t perform that action at this time.
0 commit comments