Fix MiniMax-H3 Cache-DiT BCG warning (#34848)
This commit is contained in:
+5
@@ -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.
|
||||
|
||||
Reference in New Issue
Block a user