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 = (