diff --git a/vllm_omni/worker/base.py b/vllm_omni/worker/base.py index 4433e3d02..dd431d16b 100644 --- a/vllm_omni/worker/base.py +++ b/vllm_omni/worker/base.py @@ -120,7 +120,6 @@ class OmniGPUWorkerBase(GPUWorker): self.model_runner.profile_run() self.non_torch_memory = profile_result.non_torch_increase - self.total_consumed = profile_result.total_consumed self.peak_activation_memory = profile_result.torch_peak_increase process_memory = (