Fix return

2026-02-25 18:58:39 -05:00 · 2025-12-15 16:57:02 +00:00
parent 5a74d76d41
commit dd84cc9ca2
1 changed files with 2 additions and 0 deletions
--- a/src/exo/worker/engines/mflux/pipefusion/distributed_denoising.py
+++ b/src/exo/worker/engines/mflux/pipefusion/distributed_denoising.py
@@ -319,6 +319,8 @@ class DistributedDenoising:
            -hidden_states.shape[0] :
        ]

+        return hidden_states
+
    def _async_pipeline(
        self,
        t: int,