Update WeightTransferConfig to be more standard like the others (#33989)

Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
This commit is contained in:
Harry Mellor
2026-02-06 13:15:00 +00:00
committed by GitHub
parent 6e7b1c4b59
commit 51a7bda625
2 changed files with 4 additions and 6 deletions

View File

@@ -1,13 +1,11 @@
# 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."""