Skip to content

vllm.config.weight_transfer

WeightTransferConfig dataclass

Configuration for weight transfer during RL training.

Source code in vllm/config/weight_transfer.py 
 9
10
11
12
13
14
15
@config
@dataclass
class WeightTransferConfig:
    """Configuration for weight transfer during RL training."""

    backend: Literal["nccl"] = "nccl"
    """The backend to use for weight transfer."""

backend class-attribute instance-attribute 

backend: Literal['nccl'] = 'nccl'

The backend to use for weight transfer.

__init__ 

__init__(backend: Literal['nccl'] = 'nccl') -> None