diff --git a/vllm/platforms/__init__.py b/vllm/platforms/__init__.py index badf72de4a90..a45ca988200d 100644 --- a/vllm/platforms/__init__.py +++ b/vllm/platforms/__init__.py @@ -38,7 +38,7 @@ def tpu_platform_plugin() -> str | None: # Check for Pathways TPU proxy if envs.VLLM_TPU_USING_PATHWAYS: logger.debug("Confirmed TPU platform is available via Pathways proxy.") - return "tpu_inference.platforms.tpu_jax.TpuPlatform" + return "tpu_inference.platforms.tpu_platform.TpuPlatform" # Check for libtpu installation try: