From b280f38b15e7f12034d2e1d1fb8a57d1c95a940f Mon Sep 17 00:00:00 2001 From: Wei-Yu Lin Date: Thu, 18 Dec 2025 20:07:42 +0000 Subject: [PATCH] Add TODO and remove unused codepath Signed-off-by: Wei-Yu Lin --- vllm/model_executor/model_loader/default_loader.py | 3 --- vllm/v1/worker/tpu_worker.py | 1 + 2 files changed, 1 insertion(+), 3 deletions(-) diff --git a/vllm/model_executor/model_loader/default_loader.py b/vllm/model_executor/model_loader/default_loader.py index deee2324960dd..5d24e72919693 100644 --- a/vllm/model_executor/model_loader/default_loader.py +++ b/vllm/model_executor/model_loader/default_loader.py @@ -241,9 +241,6 @@ class DefaultModelLoader(BaseModelLoader): self.load_config.pt_load_map_location, ) - if current_platform.is_tpu(): - pass - if self.counter_before_loading_weights == 0.0: self.counter_before_loading_weights = time.perf_counter() # Apply the prefix. diff --git a/vllm/v1/worker/tpu_worker.py b/vllm/v1/worker/tpu_worker.py index 085b119e12600..4c73d6c92d391 100644 --- a/vllm/v1/worker/tpu_worker.py +++ b/vllm/v1/worker/tpu_worker.py @@ -11,6 +11,7 @@ logger = init_logger(__name__) _R = TypeVar("_R") +# TODO(weiyulin) Remove this file after adding an official way to use hardware plugin if USE_TPU_INFERENCE: from tpu_inference.worker.tpu_worker import TPUWorker as TpuInferenceWorker