diff --git a/vllm/platforms/tpu.py b/vllm/platforms/tpu.py index aa5ddbe43659..04325a522f44 100644 --- a/vllm/platforms/tpu.py +++ b/vllm/platforms/tpu.py @@ -267,7 +267,7 @@ class TpuPlatform(Platform): try: - from tpu_inference.platforms.tpu_platforms import ( + from tpu_inference.platforms import ( TpuPlatform as TpuInferencePlatform, )