Fix h3 fun controlnet with comfy compiler. (#16240)

comfy_extras/nodes_minimax_h3.py

@@ -17,6 +17,7 @@ import torchaudio
 
 import nodes
 import comfy.model_management
+import comfy.model_prefetch
 import comfy.model_sampling
 import comfy.nested_tensor
 import comfy.patcher_extension
@@ -478,6 +479,7 @@ class MiniMaxH3FunControlPatch:
         self.active = self.sigma_end <= sigma <= self.sigma_start
         self.control_stream = None
         if self.active:
+            with comfy.model_prefetch.pause_malloc_graph():
                 self.prepare_control_latent(x[0].shape)
         try:
             return executor(x, timestep, context, transformer_options, **kwargs)
@@ -540,11 +542,14 @@ class MiniMaxH3FunControlBlockPatch:
         self.previous = previous
 
     def __call__(self, args, extra_args):
+        # Control state must stay outside the base block's allocation scope.
+        with comfy.model_prefetch.pause_malloc_graph():
             self.control_patch.before_block(self.block_index, args)
         if self.previous is None:
             out = extra_args["original_block"](args)
         else:
             out = self.previous(args, extra_args)
+        with comfy.model_prefetch.pause_malloc_graph():
             return self.control_patch.after_block(self.block_index, args, out)
 
     def to(self, device_or_dtype):