Remove eval

2026-01-06 10:51:20 +00:00
parent 8576f4252b
commit da823a2b02
1 changed files with 0 additions and 2 deletions
@@ -410,8 +410,6 @@ class DistributedDenoising:
                else:
                    mx.distributed.send(patch_hidden, self.rank + 1, group=self.group)

-                mx.eval(patch_hidden, encoder_hidden_states)
-
            # === PHASE 4: Single Blocks with KV Cache ===
            if self.has_single_blocks:
                # Receive from previous stage if we didn't do concatenation