A high-throughput and memory-efficient inference and serving engine for LLMs
Mirrored from https://github.com/tduncan-lum/Nvidia-Optimization-Stack---VLLM.git
Synced Nov 7, 14:00 UTC
[CI] add trainer_send_weights for MockWeightTransferEngine (#35589)
Signed-off-by: chaunceyjiang <chaunceyjiang@gmail.com>
Chauncey committed
Feb 28, 2026 at 14:47 UTC
06254d4cbb79e0a406fbf4e18d739293d5470114
1 file changed
+4
tests/entrypoints/weight_transfer/test_weight_transfer_llm.py
+4
@@ -90,6 +90,10 @@ class MockWeightTransferEngine(WeightTransferEngine[MockInitInfo, MockUpdateInfo
90
def shutdown(self) -> None:
91
MockWeightTransferEngine.shutdown_called = True
92
93
+ def trainer_send_weights(self, *args, **kwargs):
94
+ """Mock method to simulate trainer sending weights."""
95
+ pass
96
+
97
98
def mock_create_engine(config, parallel_config):
99
"""Mock factory function that returns our mock engine."""