Fix return

2026-01-06 10:51:20 +00:00
parent 5a74d76d41
commit dd84cc9ca2
1 changed files with 2 additions and 0 deletions
@@ -319,6 +319,8 @@ class DistributedDenoising:
            -hidden_states.shape[0] :
        ]

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