[Misc] add vllm_config in __init__ (#19866)

Signed-off-by: Andy Xie <andy.xning@gmail.com>
This commit is contained in:
Ning Xie 2025-06-22 11:10:46 +08:00 committed by GitHub
parent 3b1e4c6a23
commit 4c409cabc2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -509,6 +509,7 @@ class WorkerWrapperBase:
"""
self.rpc_rank = rpc_rank
self.worker: Optional[WorkerBase] = None
self.vllm_config: Optional[VllmConfig] = None
# do not store this `vllm_config`, `init_worker` will set the final
# one. TODO: investigate if we can remove this field in
# `WorkerWrapperBase`, `init_cached_hf_modules` should be