diff --git a/vllm/platforms/tpu.py b/vllm/platforms/tpu.py index aa5ddbe43..04325a522 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, )