[Feature] Bagel: Support tp+cfg parallel using mooncake transfer engine connector (#2705)

Signed-off-by: natureofnature <wzliu@connect.hku.hk>
Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com>

vllm_omni/inputs/data.py

@@ -267,6 +267,10 @@ class OmniDiffusionSamplingParams:
     cfg_text_kv_metadata: dict[str, Any] | None = None
     cfg_img_kv_metadata: dict[str, Any] | None = None
     cfg_kv_request_ids: dict[str, str] | None = None
+    cfg_active_branch: str | None = None
+    cfg_branch_roles: list[str] | None = None
+    cfg_branch_past_key_values: dict[str, Any] | None = None
+    cfg_branch_kv_metadata: dict[str, dict[str, Any]] | None = None
 
     # Component modules
     modules: dict[str, Any] = field(default_factory=dict)