[RL] Support collective rpc api to entrypoint && Support custom input output (#1646)

Signed-off-by: knlnguyen1802 <knlnguyen1802@gmail.com>
Signed-off-by: Samit <285365963@qq.com>
Co-authored-by: Samit <285365963@qq.com>

vllm_omni/diffusion/diffusion_engine.py

@@ -144,6 +144,7 @@ class DiffusionEngine:
                         prompt=prompt,
                         metrics=metrics,
                         latents=output.trajectory_latents,
+                        custom_output=output.custom_output or {},
                         multimodal_output=mm_output,
                     ),
                 ]
@@ -197,6 +198,7 @@ class DiffusionEngine:
                             prompt=prompt,
                             metrics=metrics,
                             latents=output.trajectory_latents,
+                            custom_output=output.custom_output or {},
                             multimodal_output=mm_output,
                         ),
                     )