Method unpatchify

diffsynth/models/cog_dit.py:230–232

(self, hidden_states, height, width)

    def unpatchify(self, hidden_states, height, width):
        hidden_states = rearrange(hidden_states, "B (T H W) (C P Q) -> B C T (H P) (W Q)", P=2, Q=2, H=height//2, W=width//2)
        return hidden_states
    

    def build_mask(self, T, H, W, dtype, device, is_bound):