Sumanth R Hegdeandmergify[bot] <37929162+mergify[bot]@users.noreply.github.com>
3cb83c9592
Signed-off-by: SumanthRH <sumanthrh99@gmail.com> Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>
15 lines
432 B
Python
15 lines
432 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
"""
|
|
Weight transfer engines for syncing model weights from trainers
|
|
to inference workers.
|
|
"""
|
|
|
|
from vllm.distributed.weight_transfer.base import WeightTransferEngine
|
|
from vllm.distributed.weight_transfer.factory import WeightTransferEngineFactory
|
|
|
|
__all__ = [
|
|
"WeightTransferEngine",
|
|
"WeightTransferEngineFactory",
|
|
]
|