feat: add profiling for vllm-omni

Signed-off-by: Chen Yang <2082464740@qq.com>

Author: erfgss

vllm_omni/diffusion/diffusion_engine.py

@@ -4,7 +4,7 @@
 import multiprocessing as mp
 import time
 import weakref
-from collections.abc import Callable,Iterable
+from collections.abc import Callable, Iterable
 from dataclasses import dataclass
 from typing import Any
 

vllm_omni/entrypoints/omni_stage.py

@@ -40,8 +40,8 @@
 from vllm_omni.entrypoints.stage_utils import (
     SHUTDOWN_TASK,
     OmniStageTaskType,
-    is_profiler_task,
     _to_dict,
+    is_profiler_task,
     maybe_dump_to_shm,
     set_stage_devices,
 )