[test][RL] Add sleep level 2 test and fix reload with sleep mode (#23521)

Signed-off-by: 22quinn <33176974+22quinn@users.noreply.github.com>
2025-08-25 09:25:52 -07:00
parent 0ff902f3b4
commit 2a167b2eeb
2 changed files with 32 additions and 2 deletions
--- a/vllm/v1/worker/gpu_worker.py
+++ b/vllm/v1/worker/gpu_worker.py
@@ -216,8 +216,7 @@ class Worker(WorkerBase):
        self.model_runner.update_config(overrides)

    def reload_weights(self) -> None:
-        with self._maybe_get_memory_pool_context(tag="weights"):
-            self.model_runner.reload_weights()
+        self.model_runner.reload_weights()

    @torch.inference_mode()
    def determine_available_memory(self) -> int: