From 6e27c6d86b6d9d7fcc8cb067ca2aa42d2f7f1682 Mon Sep 17 00:00:00 2001 From: Roger Wang Date: Tue, 13 May 2025 19:33:40 -0700 Subject: [PATCH] [Misc] Remove unused numpy tensor (#18084) Signed-off-by: Roger Wang --- vllm/v1/worker/gpu_model_runner.py | 1 - vllm/v1/worker/tpu_model_runner.py | 1 - 2 files changed, 2 deletions(-) diff --git a/vllm/v1/worker/gpu_model_runner.py b/vllm/v1/worker/gpu_model_runner.py index 31895cc0832aa..c2c8533c88f4d 100644 --- a/vllm/v1/worker/gpu_model_runner.py +++ b/vllm/v1/worker/gpu_model_runner.py @@ -287,7 +287,6 @@ class GPUModelRunner(LoRAModelRunnerMixin): dtype=torch.int32, device="cpu", pin_memory=self.pin_memory) - self.input_ids_np = self.input_ids_cpu.numpy() self.positions_cpu = torch.zeros(self.max_num_tokens, dtype=torch.int64, device="cpu", diff --git a/vllm/v1/worker/tpu_model_runner.py b/vllm/v1/worker/tpu_model_runner.py index 687dabee22904..b4daf5a346788 100644 --- a/vllm/v1/worker/tpu_model_runner.py +++ b/vllm/v1/worker/tpu_model_runner.py @@ -191,7 +191,6 @@ class TPUModelRunner(LoRAModelRunnerMixin): self.input_ids_cpu = torch.zeros(self.max_num_tokens, dtype=torch.int32, device="cpu") - self.input_ids_np = self.input_ids_cpu.numpy() self.positions_cpu = torch.zeros(self.max_num_tokens, dtype=torch.int32,