Add TODO and remove unused codepath

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

View File

@ -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.

View File

@ -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