[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>

tests/diffusion/test_worker_wrapper_base.py

@@ -96,6 +96,7 @@ class TestWorkerWrapperBaseInitialization:
             local_rank=0,
             rank=0,
             od_config=mock_od_config,
+            skip_load_model=False,
         )