Fix MiniMax-H3 Cache-DiT BCG warning (#34848)

This commit is contained in:
Mohammad Miadh Angkad
2026-08-14 22:07:50 +08:00
committed by GitHub
parent c939307e8a
commit 9d2f1584fa
@@ -413,6 +413,11 @@ class MiniMaxH3DenoisingStage(DenoisingStage):
current_mode = getattr(self, "_minimax_h3_cache_mode", None)
self._minimax_h3_quality = quality
if self.server_args.enable_breakable_cuda_graph:
if desired_mode is not None:
super()._maybe_enable_cache_dit(num_inference_steps, batch)
return
# H3 is monolithic-only, and the scheduler executes one worker batch at
# a time. Combined with `quality` in the dynamic-batch signature, this
# makes the process-wide hook transition safe at this batch boundary.