Skip to content

Commit df4d3a4

Browse files
authored
[TPU] Rename path to tpu platform (vllm-project#28452)
Signed-off-by: Kyuyeun Kim <[email protected]>
1 parent 9d1c474 commit df4d3a4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vllm/platforms/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ def tpu_platform_plugin() -> str | None:
3838
# Check for Pathways TPU proxy
3939
if envs.VLLM_TPU_USING_PATHWAYS:
4040
logger.debug("Confirmed TPU platform is available via Pathways proxy.")
41-
return "tpu_inference.platforms.tpu_jax.TpuPlatform"
41+
return "tpu_inference.platforms.tpu_platform.TpuPlatform"
4242

4343
# Check for libtpu installation
4444
try:

0 commit comments

Comments
 (0)