From 39c4a4cdb592d953df75400e17b5bc899d54329d Mon Sep 17 00:00:00 2001 From: Alexander Matveev Date: Tue, 28 Jan 2025 23:01:54 +0000 Subject: [PATCH] review comments Signed-off-by: Alexander Matveev --- vllm/v1/worker/tpu_model_runner.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/vllm/v1/worker/tpu_model_runner.py b/vllm/v1/worker/tpu_model_runner.py index 5baa086507..fc659c864a 100644 --- a/vllm/v1/worker/tpu_model_runner.py +++ b/vllm/v1/worker/tpu_model_runner.py @@ -556,9 +556,6 @@ class TPUModelRunner(ModelRunnerBase): assert self.model is not None self.model(token_ids, position_ids, None, kv_caches) - def profile_run(self) -> None: - raise NotImplementedError() - def capture_model(self) -> None: """Compile the model."""