xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-05-23 18:17:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add TODO and remove unused codepath

Browse Source 
Signed-off-by: Wei-Yu Lin <weiyulin@google.com>
This commit is contained in:
Wei-Yu Lin 2025-12-18 20:07:42 +00:00
parent 603a1bf9bc
commit b280f38b15
2 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
vllm/model_executor/model_loader/default_loader.py
View File

@ -241,9 +241,6 @@ class DefaultModelLoader(BaseModelLoader):
self.load_config.pt_load_map_location, self.load_config.pt_load_map_location,
) )
if current_platform.is_tpu():
pass
if self.counter_before_loading_weights == 0.0: if self.counter_before_loading_weights == 0.0:
self.counter_before_loading_weights = time.perf_counter() self.counter_before_loading_weights = time.perf_counter()
# Apply the prefix. # Apply the prefix.

1
vllm/v1/worker/tpu_worker.py
View File

@ -11,6 +11,7 @@ logger = init_logger(__name__)
_R = TypeVar("_R") _R = TypeVar("_R")
# TODO(weiyulin) Remove this file after adding an official way to use hardware plugin
if USE_TPU_INFERENCE: if USE_TPU_INFERENCE:
from tpu_inference.worker.tpu_worker import TPUWorker as TpuInferenceWorker from tpu_inference.worker.tpu_worker import TPUWorker as TpuInferenceWorker