b6d2b4fc25
Signed-off-by: Ke Wen <kwen@nvidia.com> Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com> Co-authored-by: aoshen02 <aoshen@inferact.ai>
18 lines
529 B
Python
18 lines
529 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
from typing import Literal
|
|
|
|
from vllm.config.utils import config
|
|
|
|
|
|
@config
|
|
class WeightTransferConfig:
|
|
"""Configuration for weight transfer during RL training."""
|
|
|
|
backend: Literal["nccl", "ipc", "sparse_nccl", "sharded_rdt", "nccl_m2n"] | str = (
|
|
"nccl"
|
|
)
|
|
"""The backend to use for weight transfer. Validated against the
|
|
`WeightTransferEngineFactory` registry at engine creation time.
|
|
"""
|