c1858b7ec8
Signed-off-by: ahao-anyscale <ahao@anyscale.com> Signed-off-by: Aaron Hao <ahao@anyscale.com> Co-authored-by: SumanthRH <sumanthrh99@gmail.com>
16 lines
408 B
Python
16 lines
408 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
from dataclasses import dataclass
|
|
from typing import Literal
|
|
|
|
from vllm.config.utils import config
|
|
|
|
|
|
@config
|
|
@dataclass
|
|
class WeightTransferConfig:
|
|
"""Configuration for weight transfer during RL training."""
|
|
|
|
backend: Literal["nccl"] = "nccl"
|
|
"""The backend to use for weight transfer."""
|