[Diffusion] Revamp Rollout Log-Prob Support with SDE/CPS for RL Post-Training#21204

Open

Rockdu wants to merge 10 commits intosgl-project:mainfrom

MikukuOvO:feat/rollout-logprob-support-revamp-20260323

Commits on Mar 23, 2026

[diffusion] feat: add rollout log_prob with flow-matching SDE/CPS support

MikukuOvO
authored and
Rockdu
committed
[diffusion] add rollout log-prob engine with SDE/CPS/ODE support
Rockdu
committed
[diffusion] integrate rollout into scheduler step and denoising pipeline
Rockdu
committed
[diffusion] expose rollout parameters in SamplingParams and API layer
Rockdu
committed
[diffusion] adapt zimage DiT for rollout SP compatibility
Rockdu
committed
[diffusion] add unit tests for rollout step alignment with FlowGRPO
Rockdu
committed
[diffusion] fix lint, align log_prob test with FlowGRPO, fix CLI defaults
Rockdu
committed
[diffusion] remove rollout params from OpenAI-compatible API layer
Rockdu
committed
[diffusion] rename test_scheduler_rollout_ode_unit -> test_scheduler_rollout_unit
Rockdu
committed

Commits on Mar 25, 2026

[diffusion] fix rollout zero noise_level
Rockdu
committed