[1/N] DP-Refactor: move communicators into tokenizer_communicator_mixin (#10028)

This commit is contained in:
Liangsheng Yin
2025-09-08 16:27:37 +08:00
committed by GitHub
parent bfd7a18d8d
commit 78f139812a
5 changed files with 503 additions and 459 deletions

View File

@@ -6,7 +6,7 @@ from torch.distributed.device_mesh import DeviceMesh
from torch.distributed.tensor import DTensor
from sglang.srt.entrypoints.engine import Engine
from sglang.srt.managers.tokenizer_manager import UpdateWeightsFromTensorReqInput
from sglang.srt.managers.io_struct import UpdateWeightsFromTensorReqInput
from sglang.srt.model_executor.model_runner import LocalSerializedTensor
from sglang.srt.utils import MultiprocessingSerializer