[Bugfix] Fix missing memory metrics info for bagel (ar->dit diffusion) - #4477

[Bugfix] Fix missing memory metrics info for bagel (ar->dit diffusion)#4477

hsliuustc0106 merged 3 commits into

vllm-project:mainvllm-project/vllm-omni:mainfrom

natureofnature:bagel-profiler-stage-metricsnatureofnature/vllm-omni:bagel-profiler-stage-metricsCopy head branch name to clipboard