From f4640d9a2e97b311e0c8c3ae4ba793d9454f7cfa Mon Sep 17 00:00:00 2001
From: Ting FU <semmer@live.cn>
Date: Tue, 28 Jul 2026 20:13:33 +0800
Subject: [PATCH] [BugFix] Fix 'GPUARWorker' attribute error (#5478)

Signed-off-by: Semmer <semmer@live.cn>
---
 vllm_omni/worker/base.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/vllm_omni/worker/base.py b/vllm_omni/worker/base.py
index dd431d16b..4433e3d02 100644
--- a/vllm_omni/worker/base.py
+++ b/vllm_omni/worker/base.py
@@ -120,6 +120,7 @@ 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 = (