[Refactor]: clean up diffusion scheduler interfaces (#5216)

Signed-off-by: Huang, Zeyu <11222265+fhfuih@users.noreply.github.com>
Signed-off-by: Zeyu Huang | 黃澤宇 <11222265+fhfuih@users.noreply.github.com>
Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>
Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
authored and GitHub committed 2026-07-22 14:05:49 +08:00
1 parent e2115fb642
commit e39f4dddfb
31 files changed
+319 -338

No files matched your search

@@ -329,7 +329,7 @@ class MultiprocDiffusionExecutor(DiffusionExecutor):
try:
result = self.collective_rpc(
"execute_model",
args=(req, self.od_config, scheduler_output.kv_prefetch_jobs),
args=(req, self.od_config, scheduler_output.kv_prefetch_job),
unique_reply_rank=0,
exec_all_ranks=True,
)