Skip to content

Commit

Permalink
[Bugfix] Fix AssertError in vllm_ascend/worker.py
Browse files Browse the repository at this point in the history
Signed-off-by: wwfu <wangwanfu@huawei.com>
  • Loading branch information
wwfu committed Feb 26, 2025
1 parent ca807ce commit 352140c
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions vllm_ascend/worker.py
Original file line number Diff line number Diff line change
Expand Up @@ -85,9 +85,6 @@ def __init__(
self.distributed_init_method = distributed_init_method
self.is_driver_worker = is_driver_worker

if is_driver_worker:
assert rank % self.parallel_config.tensor_parallel_size == 0, \
"Driver worker should be rank 0 of tensor parallel group."
if self.model_config.trust_remote_code:
# note: lazy import to avoid importing torch before initializing
from vllm.utils import init_cached_hf_modules
Expand Down

0 comments on commit 352140c

Please sign in to comment.