Restore Literal for WeightTransferConfig.backend (#43183)

Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
This commit is contained in:
Harry Mellor
2026-05-28 09:39:41 +00:00
committed by GitHub
parent a04afd76aa
commit 2a781756a1
+3 -1
View File
@@ -1,5 +1,7 @@
# SPDX-License-Identifier: Apache-2.0
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
from typing import Literal
from vllm.config.utils import config
@@ -7,7 +9,7 @@ from vllm.config.utils import config
class WeightTransferConfig:
"""Configuration for weight transfer during RL training."""
backend: str = "nccl"
backend: Literal["nccl", "ipc"] | str = "nccl"
"""The backend to use for weight transfer. Validated against the
`WeightTransferEngineFactory` registry at engine creation time.
"""