[Feature] Add MiniMax H3 res_multistep sampler (#8378)

Signed-off-by: nodeeeeee <zhangkai.nodeee@gmail.com>
This commit is contained in:
nodeeeeee authored and GitHub committed 2026-10-04 07:26:19 +00:00
1 parent 148cb943ce
commit b65f97bc48
5 files changed
+166 -23

No files matched your search

@@ -102,7 +102,11 @@ def _make_state(request_id: str, model, branch, video_rows, audio_rows, sigmas_v
state.latents = video_rows.clone()
state.timesteps = torch.tensor([1.0 - sigma for sigma in sigmas_video[:-1]], dtype=torch.float32)
state.step_index = 0
from vllm_omni.diffusion.models.minimax_h3.sampling import create_h3_sample_solver
state.extra = {
mod._STEP_VIDEO_SOLVER: create_h3_sample_solver("euler", sigmas_video),
mod._STEP_AUDIO_SOLVER: create_h3_sample_solver("euler", sigmas_audio),
mod._STEP_BRANCH: branch,
# Co-batched requests must share one DiT instance, or denoise_step()
# treats the batch as mixed-task and falls back to one forward each.